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