力扣 leetcode 86. 分隔链表 （python）

Topic：

给你一个链表和一个特定值 x ，请你对链表进行分隔，使得所有小于 x 的节点都出现在大于或等于 x 的节点之前。

你应当保留两个分区中每个节点的初始相对位置。

Example：

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

Solution：

整体的思路就是将链表按原序拆分为小的和大的两个链表，最后合并

首先先设置两个哑节点（便于边界的判断）
之后遍历整个链表
若head的val小于x，则small的下一位变为head
small指针转换为下一位（此时head的val）

同理
若head的val大于或等于于x，则big的下一位变为head
big指针指向下一位（此时head的val）
这样两个链表已经按照原来的顺序
分别将小的保存为了small
大的保存为了big

遍历结束后
将 big 的 next 指针清空（防止指向原来初始链表的节点）
同时将 small 的 next 指向L2的next 的节点（big链表真正的头节点）
最后返回L1的next指针

Code：

class Solution:
    def partition(self, head: ListNode, x: int) -> ListNode:
        L1 = small = ListNode(0)
        L2 =  big = ListNode(0)
        
        while head:
            if head.val < x:
                small.next = head
                small = small.next
            else:
                big.next = head
                big = big.next
            
            head = head.next

        big.next = None
        small.next = L2.next
        
        return L1.next

Result：