partition-list-leetcode-C++

最新推荐文章于 2024-07-04 10:24:50 发布

原创最新推荐文章于 2024-07-04 10:24:50 发布 · 593 阅读

0 ·

CC 4.0 BY-SA版权

leetcode解题报告专栏收录该内容

2 篇文章

订阅专栏

本文介绍了一种链表分隔算法，该算法可以将链表中的节点根据给定的值x进行分隔，使得所有小于x的节点位于大于等于x的节点之前，同时保持各分区内的相对顺序不变。

题目：

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,
Given1->4->3->2->5->2and x = 3,
return1->2->2->4->3->5.

//思路：开辟两个新链表，将大于x的值放在一个链表上，小于x的值放在一个链表上

贴上AC代码如下：

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
    
public:
    ListNode *partition(ListNode *head, int x) {
        if(head==NULL)
             return head;
        ListNode* big=new ListNode(-1);//定义两个新的链表
        ListNode* small=new ListNode(-1);
        ListNode* bigCur=big;//记录每个新链表的当前末尾的非空节点
        ListNode* smallCur=small;
        ListNode* p=head;
        while(p!=NULL)
        {
            if(p->val>=x)//添加到大链表中
            {
                bigCur->next=p;
                bigCur=p;
            }    
            else //添加到小链表中
            {
                smallCur->next=p; 
                smallCur=p;
            }    
            p=p->next;
        }
        bigCur->next=NULL;
        smallCur->next=big->next;
        return small->next;
    }
};