LeetCode 每日一题 2021-1-3 （分隔链表）

86. 分隔链表

难度中等

给你一个链表和一个特定值 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;
}