给你一个链表的头节点 head
和一个特定值 x
,请你对链表进行分隔,使得所有 小于 x
的节点都出现在 大于或等于 x
的节点之前。
你应当 保留 两个分区中每个节点的初始相对位置。
示例 1:
输入:head = [1,4,3,2,5,2], x = 3
输出:[1,2,2,4,3,5]
示例 2:
输入:head = [2,1], x = 2
输出:[1,2]
提示:
[0, 200]
内-100 <= Node.val <= 100
-200 <= x <= 200
把比目标值小的节点存一个链表里,把比目标值大的节点另一个链表里
把存比目标值大的节点的链表接到存比目标值小的节点的链表的后面
如果大于,那么pre指向当前节点,然后继续遍历
如果小于,那么看pre所指向的节点是否是小区间的尾节点
如果是,那么pre指向当前节点,然后继续遍历
如果不是 ,(1)我们让pre指向的那个节点的下一个节点变为为当前节点的下一个节点
(2)当前节点指向小区间尾节点的下一个节点,然后小区间的尾节点再指向当节点 (3)小区间的尾节点向后移动一个节点,下一次要遍历的节点为pre所指向节点的下一个节点
class Solution {
public:
ListNode* partition(ListNode* head, int x) {
if(head==nullptr||head->next==nullptr){
return head;
}
//遍历一遍链表拆成两个链表
ListNode* head1=nullptr;
ListNode* head2=nullptr;
ListNode* tail1=nullptr;
ListNode* tail2=nullptr;
while(head){
if(head->valnext=head;
tail1=tail1->next;
}
}
else{
if(head2==nullptr){
head2=tail2=head;
}
else{
tail2->next=head;
tail2=tail2->next;
}
}
head=head->next;
}
if(tail1){
tail1->next=nullptr;
}
if(tail2){
tail2->next=nullptr;
}
if(tail1){
tail1->next=head2;
return head1;
}
else{
return head2;
}
}
};
class Solution {
public:
ListNode* partition(ListNode* head, int x) {
if(head ==nullptr || head->next==nullptr){
return head;
}
ListNode* dummyhead = new ListNode(0,head);
ListNode* prevtail = dummyhead,*prev = dummyhead,*curr = head;
while(curr){
if(curr->val < x){
if(prev != prevtail){
prev->next = curr->next;
curr->next = prevtail->next;
prevtail->next = curr;
prevtail = prevtail->next;
curr = prev->next;
}else{
prev = prevtail = curr;
curr = curr->next;
}
}else{
prev = curr;
curr = curr->next;
}
}
return dummyhead->next;
}
};