难度中等
给你一个链表和一个特定值 x
,请你对链表进行分隔,使得所有小于 x
的节点都出现在大于或等于 x
的节点之前。
你应当保留两个分区中每个节点的初始相对位置。
思路:
分隔链表思路:创建四个指针
两个分别指向小于指定值的开始位置 大于等于指定值的开始位置
一个历遍一次链表,辅助分别连接上述和指定值比较的两个部分,并从中间断开
一个用于构建新小于指定值的链表 ,并连接大于指定值的开始位置
代码:
// 分隔链表思路:创建四个指针
// 两个指向小于指定值的开始位置 大于等于指定值的开始位置
// 一个历遍一次链表,辅助连接上述两部分,并断开
// 一个用于构建新小于指定值的链表 ,并连接大于指定值的开始位置
ListNode* partition(ListNode* head, int x) {
if(head==NULL || head->next==NULL) return head;
ListNode* cur=head,* ansHead=NULL,* ansCur=ansHead,* node=NULL;
if(head->val<x){ ansHead=head;ansCur=head;}
else node=head;
// 如果第一个是小于指定值的数据 找到大于指定值的数据
while(ansCur!=NULL){
if(ansCur->next!=NULL && ansCur->next->val <x) ansCur=ansCur->next;
else {
cur=ansCur->next;
node=ansCur->next;
break;
}
}
// 分布链接两个子链表
while(cur!=NULL && cur->next!=NULL){
if(cur->next->val < x){
if(ansHead==NULL) { ansHead=cur->next;ansCur=cur->next;}
else { ansCur->next=cur->next;ansCur=ansCur->next;} // 已连接
cur->next=cur->next->next;
}
else {if(node==NULL) node=cur->next; cur=cur->next;}
}
// 断开子链表间的链接
if(cur!=NULL) cur->next=NULL;
// 重新链接两个子数据
if(ansHead!=NULL) ansCur->next=node;
else return node; // 不存在小于的情况
return ansHead;
}