分隔链表

链表分隔算法

最新推荐文章于 2021-09-22 19:07:20 发布

原创最新推荐文章于 2021-09-22 19:07:20 发布 · 166 阅读

0 ·

CC 4.0 BY-SA版权

python 同时被 2 个专栏收录

143 篇文章

订阅专栏

算法

57 篇文章

订阅专栏

本文介绍了一种链表分隔算法，该算法接收一个链表和一个特定值x作为输入，将链表中小于x的节点置于大于或等于x的节点之前，同时保持两个分区中节点的初始相对位置不变。

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

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

示例:

输入: head = 1->4->3->2->5->2, x = 3
输出: 1->2->2->4->3->5

    def partition(self, head, x):
        pre = None
        res = right = head
        while head:
            if head.val < x:
                if right is head:
                    pre, head, right = head, head.next, head.next
                else:
                    if pre is None:
                        head.next, res, head, pre, right.next = res, head, head.next, head, head.next
                        continue
                    head.next, pre.next, right.next, head, pre = pre.next, head, head.next, head.next, head
            else:
                right, head = head, head.next
        return res