86. 分隔链表

86. 分隔链表

原始题目链接：https://leetcode-cn.com/problems/partition-list/

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

解题思路：

以x大小为界，将原始链表分为大小两个链表，然后再拼接一起即可，具体实现看代码及注释。

代码实现：

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, val=0, next=None):
#         self.val = val
#         self.next = next
class Solution:
    def partition(self, head: ListNode, x: int) -> ListNode:
        # 判空
        if head is None:
            return head
        
        # 申请两个节点，即两个dummy
        dummy1 = ListNode(-1)
        dummy2 = ListNode(-1)

        # 分别遍历指向head较小的节点和较大的节点
        # 形成以x为界限的大小两个链表
        small = dummy1
        large = dummy2

        while head:
            if head.val < x:
                small.next = head
                small = small.next
            else:
                large.next = head
                large = large.next
            
            head = head.next
        
        # 将大、小链表进行拼接
        # 小的链表的尾节点的next指针指向大链表的头结点
        small.next = dummy2.next
        # 大链表的尾节点的next指针指向空，表示整个拼接后链表的尾节点
        large.next = None

        # 返回小链表的头结点就是最后拼接后链表的头结点
        return dummy1.next

参考文献：
https://leetcode-cn.com/problems/partition-list/solution/fen-ge-lian-biao-by-leetcode-solution-7ade/