LeetCode Partition List(链表分段)

最新推荐文章于 2024-07-10 15:20:53 发布

原创最新推荐文章于 2024-07-10 15:20:53 发布 · 1k 阅读

1 ·

CC 4.0 BY-SA版权

OJ 专栏收录该内容

986 篇文章

订阅专栏

本文介绍了一种单链表的分区算法实现，该算法类似于快速排序的分段处理方式，可以将链表中小于给定值x的所有元素移动到大于等于x的元素前面。通过增加头结点简化操作流程，并特别注意避免出现环的情况。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题意：给出一个单链表，有参数的数x。将比x小的数排在比x大的数据后面

思路：类似于快速排序的分段处理。为了简单操作，在单链表前加个头结点。注意，在处理时，如果当前处理结点与被替换结点为同一结点结，两个都下移，否则会出现环

代码如下：

class Solution
{
    public ListNode partition(ListNode head, int x)
    {
        if (null == head || null == head.next) return head;

        ListNode dummy = new ListNode(0);
        dummy.next = head;

        ListNode p = dummy, cur = dummy;

        while (cur != null && cur.next != null)
        {
            if (cur.next.val < x)
            {
                if (cur == p)
                {
                    p = p.next;
                    cur = cur.next;
                    continue;
                }

                ListNode tmp = cur.next;
                ListNode next = p.next;
                cur.next = cur.next.next;
                p.next = tmp;
                tmp.next = next;
                p = p.next;
                continue;
            }

            cur = cur.next;
        }

        return dummy.next;
    }
}