0x01.问题
给定一个链表和一个特定值 x,对链表进行分隔,使得所有小于 x 的节点都在大于或等于 x 的节点之前。
你应当保留两个分区中每个节点的初始相对位置。
输入示例: head = 1->4->3->2->5->2, x = 3
输出示例:1->2->2->4->3->5
C++结构体:
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
C++函数形式: ListNode* partition(ListNode* head, int x)
0x02.简要分析
要保证前面的元素都比x
小,后面的元素都比x
大,且相对的位置不变,我们只需要使用两个指针,小的元素往第一个里面添加,大的往第二个里面添加,最后再合并起来就可以了。
需要注意的地方:
head
节点遍历完后,需要对第二个指针的next
初始化,因为这将是返回链表的尾部。- 用哑节点处理两个链表。
- 接的时候也要注意接的是哑节点的
next
。
0x03.解决代码
class Solution {
public:
ListNode* partition(ListNode* head, int x) {
ListNode*minx=new ListNode(0);
ListNode*maxx=new ListNode(0);
ListNode*before=minx;
ListNode*after=maxx;
while(head){
if(head->val<x){
before->next=head;
before=before->next;
}
else{
after->next=head;
after=after->next;
}
head=head->next;
}
after->next=NULL;
before->next=maxx->next;
return minx->next;
}
};
ATFWUS --Writing By 2020–03–24