LeetCode #86 - Partition List

最新推荐文章于 2022-10-11 08:38:46 发布

原创最新推荐文章于 2022-10-11 08:38:46 发布 · 169 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

625 篇文章

订阅专栏

本文介绍了一种链表分隔算法，该算法能够将链表中小于特定值的节点置于大于或等于该值的节点之前，同时保持各部分内节点的原始顺序。通过创建两个辅助链表来分别存储小于目标值的节点和大于等于目标值的节点，最终将这两个链表合并为一个有序的链表。

题目描述：

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.

Example:

Input: head = 1->4->3->2->5->2, x = 3
Output: 1->2->2->4->3->5

将链表中小于目标值的节点放在大于或等于目标值的节点前，同时不改变两个部分的相对顺序。

定义两个链表，分别存储两部分，最后将它们连接起来即可。

class Solution {
public:
    ListNode* partition(ListNode* head, int x) {
        ListNode* head1=new ListNode(0);
        ListNode* head2=new ListNode(0);
        ListNode* p=head1;
        ListNode* q=head2;
        while(head!=NULL)
        {
            if(head->val<x) 
            {
                ListNode* temp=new ListNode(head->val);
                p->next=temp;
                p=p->next;
                
            }
            else if(head->val>=x) 
            {
                ListNode* temp=new ListNode(head->val);
                q->next=temp;
                q=q->next;
                
            }
            head=head->next;
        }
        p->next=head2->next;
        q->next=NULL;
        return head1->next;
    }
};