题目

给你一个链表的头节点 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