[leetcode-链表]86. 分隔链表

给定一个链表和一个特定值 x，对链表进行分隔，使得所有小于 x 的节点都在大于或等于 x 的节点之前。

你应当保留两个分区中每个节点的初始相对位置。

示例:

输入: head = 1->4->3->2->5->2, x = 3
输出: 1->2->2->4->3->5

 

大数结点指针总是指向后续结点，遇到小数结点，则前移到小数结点的尾结点

//====================================================================================

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     struct ListNode *next;
 * };
 */
struct ListNode* partition(struct ListNode* head, int x) {
    
    if(head==NULL || head->next==NULL)
    {
        return head;
    }
    
    struct ListNode *pcur=head;
    struct ListNode *bigHead=NULL, *bigNext=NULL;
    struct ListNode *smallNext=NULL;
    
    while(pcur)
    {
        if(pcur->val < x) //当前结点为小数结点
        {
            if(bigNext==NULL) //如果大数头结点为空，则说明遍历过的所有结点都是小节点
            {
                smallNext=pcur;   
            }
            else //如果大数头结点不为空，则需要考虑大数结点的前进
            {
                if(smallNext==NULL) //如果小数结点为空，则找到小数结点的头结点
                {
                    smallNext = pcur;
                    head = pcur;
                }
                else //如果小数结点不为空，则挂接该数
                {
                    smallNext->next = pcur;
                    smallNext=pcur;
                }
                
                bigNext->next=pcur->next; //因为当前结点是小数结点，所以需要让大数结点的下一个节点指向当前结点下一个节点
            }
        }
        else //如果当前结点是大数结点
        {
            if(bigHead==NULL)
            {
                bigHead=pcur;
            }

            bigNext=pcur; //大数结点前进到当前结点
        }
        
        pcur=pcur->next; //当前结点前进一步
        
    }
    
    if(smallNext) //如果小数结点尾结点非空，则挂接大数结点的头结点
        smallNext->next = bigHead;
    
    return head;
}