每日leetcode——86. 分隔链表

题目

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

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

输入：head = [1,4,3,2,5,2], x = 3
输出：[1,2,2,4,3,5]

输入：head = [2,1], x = 2
输出：[1,2]

一开始题目都没读懂...
其实题目的意思就是：比如示例中的数组，小于3的都放到前面，大于等于3的都放到后面，并且不改变相对顺序。

思路

维护两个链表：大链表、小链表
大链表用来存放大于等于x的部分
小链表用来存放小于x的部分
最后将小链表和大链表拼接起来，就是最终答案

def partition(head: ListNode, x: int) -> ListNode:
    bigHead, smallHead = ListNode(-1), ListNode(-1)
    bigTail,smallTail = bigHead,smallHead

    while head:
        if head.val < x:
            smallTail.next = head
            smallTail = smallTail.next
        else:
            bigTail.next = head
            bigTail = bigTail.next
        head = head.next

    smallTail.next = bigHead.next
    bigTail.next = None
    return smallHead.next