假设我们有一个单链表和另一个值k。我们必须对节点进行排列,以使所有值小于k的节点排在最前面,然后所有值等于k的节点排在最后,最后是其他节点。约束是节点的相对顺序应保持相同。
因此,如果输入像L = [4,3,6,6,6,6,10,8] k = 6,那么输出将是[4,3,6,6,6,6,10,8,]
为了解决这个问题,我们将遵循以下步骤-
less_head:=创建一个值等于0的链表节点
少:= less_head
equal_head:=创建一个值等于0的链表节点
等于:= equal_head
Greater_head:=创建一个值等于0的链表节点
更大:= Greater_head
cur:=节点
当cur不为空时,
next of equal:=创建一个值等于cur值的链表节点
等于:=下一个等于
更大的下一个:=创建一个值等于cur值的链表节点
更大:=更大的下一个
下一个:=创建一个值等于cur值的链表节点
少:=少
如果cur的值<k,则
否则,当cur的值> k时,则
除此以外,
cur:=下一个cur
少的下一个:= equal_head的下一个
等于的下一个:= Greater_head的下一个
返回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))
[4, 3, 6, 6, 6, 10, 8], 6
输出结果
[4, 3, 6, 6, 6, 10, 8, ]