leetcode 86. Partition List

最新推荐文章于 2025-11-29 20:54:03 发布

原创最新推荐文章于 2025-11-29 20:54:03 发布 · 236 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

leetcode 专栏收录该内容

12 篇文章

订阅专栏

本文介绍了一种链表分隔算法，该算法基于快速排序思想，实现将链表中小于给定值x的所有节点置于大于等于x的节点之前，同时保持各分区内部节点的原始相对顺序不变。

Description:
Given a linked list and a value x, partition it such that all nodes less than x come before nodes greater than or equal to x.

You should preserve the original relative order of the nodes in each of the two partitions.

For example,
Given 1->4->3->2->5->2 and x = 3,
return 1->2->2->4->3->5.

采用快排的思想，将比x小的数都放在左边，其他都放在右边，保持其他原有的相对顺序不变。

Solution:
1、我们需要保存4个变量，分别是快速排序中左半部分和右半部分的开头和结尾
2、遍历完之后，是需要将左半部分的结尾和右半部分的开头相连
3、这里值得注意的是，要将右半部分的结尾的next指针设为NULL，不然会被判TTL。

Code:

class Solution {
public:
    ListNode* partition(ListNode* head, int x) {
        ListNode* temp=head;
        ListNode* newhead=NULL;//左边最前面的节点，新链头
        ListNode* mid1=NULL;//左边最后面的节点
        ListNode* mid2=NULL;//右边最前面的节点
        ListNode* newtail=NULL;//右边最后面的节点,新链尾
        while(temp)
        {
            ListNode* temp1=temp->next;
            if(temp->val<x)
            {
                if(!newhead)
                    newhead=temp;
                else
                    mid1->next=temp;
                mid1=temp;
            }
            else
            {
                if(!mid2)
                    mid2=temp;
                else
                    newtail->next=temp;
                newtail=temp;
            }
            temp=temp1;
        }
        if(newtail)
            newtail->next=NULL;
        if(mid1)
            mid1->next=mid2;//需要判断mid1是否为空
        else
            newhead=mid2;
        return newhead;
    }
};