LeetCode 86. 分隔連結串列

題目內容

86. 分隔連結串列
    給你一個連結串列和一個特定值 x ，請你對連結串列進行分隔，使得所有小於 x 的節點都出現在大於或等於 x 的節點之前。

你應當保留兩個分割槽中每個節點的初始相對位置。

示例：

輸入：head = 1->4->3->2->5->2, x = 3
輸出：1->2->2->4->3->5

解題思路

copy別人的打趣段子：天晴了雨停了我又覺得我行了。
使用兩個連結串列一個記錄小於x元素，一個記錄大於x的元素，將第二個連結串列加在第一個的尾部得到結果。有這樣一個簡單的思路解題不難。
關鍵在於細節的空指標與越界異常處理。

解題程式碼

class Solution {
public:
    ListNode* partition(ListNode* head, int x) {
        ListNode* n1=nullptr,*n2=nullptr;
        ListNode *res=nullptr;ListNode *t=nullptr;
        while(head!=NULL)
        {
            if(head->val<x)
            {
                if(n1){n1->next=new ListNode;n1=n1->next;n1->val=head->val;n1->next=nullptr;}
                else {res=n1=new ListNode;res->val=head->val;res->next=nullptr;}
            }
            else
            {
                if(n2){n2->next=new ListNode;n2=n2->next;n2->val=head->val;n2->next=nullptr;}
                else {t=n2=new ListNode;t->val=head->val;t->next=nullptr;}
            }
            head=head->next;
        }
        if(!res)return t;
        n1->next=t;return res;
    }
};