Python 中根据 k 值排列链表节点的程序

pythonserver side programmingprogramming

假设我们有一个单链表和另一个 k 值。我们必须排列节点，使所有值小于 k 的节点排在最前面，所有值等于 k 的节点排在后面，最后是其他节点。约束是节点的相对顺序应保持不变。

因此，如果输入为 L = [4, 3, 6, 6, 6, 10, 8] k = 6，则输出将为 [4, 3, 6, 6, 6, 10, 8, ]

为了解决这个问题，我们将遵循以下步骤 −

less_head := 创建一个值与 0 相同的链接列表节点
less := less_head
equal_head := 创建一个值与 0 相同的链接列表节点
equal := equal_head
greater_head := 创建一个值与 0 相同的链接列表节点
greater := greater_head
cur :=节点
当 cur 不为空时，执行
- 如果 cur 的值 < k，则
  - less 的下一个 := 创建一个链接列表节点，其值与 cur 的值相同
  - less := less 的下一个
- 否则，当 cur 的值 > k，则
  - next of greater := 创建一个链表节点，其值与 cur 的值相同
  - greater := next of greater
- 否则，
  - next of equal := 创建一个链表节点，其值与 cur 的值相同
  - equal := next of equal
- cur := next of cur
next of less := next of equal_head
next of equal := next of greater_head
返回 next of less_head

让我们看看下面的实现以便更好地理解 −

示例

class ListNode:
   def __init__(self, data, next = None):
      self.val = data
      self.next = next
def make_list(elements):
   head = ListNode(elements[0])
   for element in elements[1:]:
      ptr = head
      while ptr.next:
         ptr = ptr.next
      ptr.next = ListNode(element)
   return head
def print_list(head):
   ptr = head
   print('[', end = "")
   while ptr:
      print(ptr.val, end = ", ")
      ptr = ptr.next
   print(']')
class Solution:
   def solve(self, node, k):
   less_head = less = ListNode(0)
   equal_head = equal = ListNode(0)
   greater_head = greater = ListNode(0)
   cur = node
   while cur:
      if cur.val < k:
         less.next = ListNode(cur.val)
         less = less.next
      elif cur.val > k:
         greater.next = ListNode(cur.val)
         greater = greater.next
      else:
         equal.next = ListNode(cur.val)
         equal = equal.next
         cur = cur.next
         less.next = equal_head.next
         equal.next = greater_head.next
      return less_head.next
ob = Solution()
L = make_list([4, 3, 6, 6, 6, 10, 8])
k = 6
print_list(ob.solve(L, k))