Leetcode 86. 分隔链表（Partition List）

Leetcode 86.分隔链表

1 题目描述（Leetcode题目链接）

  给定一个链表和一个特定值 x，对链表进行分隔，使得所有小于 x 的节点都在大于或等于 x 的节点之前。
  你应当保留两个分区中每个节点的初始相对位置。

输入: head = 1->4->3->2->5->2, x = 3
输出: 1->2->2->4->3->5

2 题解

  两个分区每个节点不改变相对位置，那么可以用快慢指针来做，慢指针总是指向小于$x$的最后一个数，快指针从慢指针的下一个位置出发：

• 如果快指针的下一个值小于$x$则将这个节点插入到慢指针之后，慢指针向前移动一位；
• 如过快指针的下一个值大于等于$x$则快指针向前移动一位。

可以加入一个虚拟头结点方便处理边界。

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution:
    def partition(self, head: ListNode, x: int) -> ListNode:
        if not head or not head.next:
            return head
        dummy = ListNode(0)
        dummy.next = head
        fast = dummy
        while fast.next and fast.next.val < x: #找到最后一个小于x的节点
            fast = fast.next
        slow, fast = fast, fast.next 

        while fast and fast.next:
            if fast.next.val < x:
                temp = fast.next
                fast.next = temp.next
                temp.next = slow.next
                slow.next = temp
                slow = slow.next
            else:
                fast = fast.next
        return dummy.next