关于数据结构与算法:每日leetcode86-分隔链表

27次阅读

共计 584 个字符,预计需要花费 2 分钟才能阅读完成。

题目

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

正文完
 0