给你一个链表和一个特定值 x ,请你对链表进行分隔,使得所有小于 x 的节点都出现在大于或等于 x 的节点之前。
你应当保留两个分区中每个节点的初始相对位置。
示例:
输入:head = 1->4->3->2->5->2, x = 3
输出:1->2->2->4->3->5
思路:
将链表分为两部分,一部分小于x,另一部分大于等于x的。遍历head,完成后将两个链表接起来即可。
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode() : val(0), next(nullptr) {}
* ListNode(int x) : val(x), next(nullptr) {}
* ListNode(int x, ListNode *next) : val(x), next(next) {}
* };
*/
class Solution {
public:
ListNode* partition(ListNode* head, int x) {
ListNode* h1=new ListNode(-1);
ListNode* h2=new ListNode(-1);
ListNode *p1=h1, *p2=h2;
while(head){
if(head->val < x){
p1->next=head;
p1=p1->next;
}
else{
p2->next=head;
p2=p2->next;
}
head=head->next;
}
p1->next=h2->next;
p2->next=nullptr;
auto res=h1->next;
delete h1,h2;
return res;
}
};