1. 程式人生 > 實用技巧 >20.1.3 86. 分隔連結串列

20.1.3 86. 分隔連結串列

題目

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

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

示例:

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

思路

  1. 思路很簡單,雙指標,兩個偽頭部節點,把小於x的串起來,大於等於x的穿起來,兩個合併即可
  2. 雖然我也沒跑出來。。。之前做過一次也沒跑出來,忘了把p2->next設空

程式碼

class Solution {
public:
    ListNode* partition(ListNode* head, int x) {
        ListNode* dummyHead1 = new ListNode(-1);
        ListNode* dummyHead2 = new ListNode(-1);
        ListNode* p1 = dummyHead1, *p2 = dummyHead2, *curr = head;
        while(curr){
            if(curr->val < x) {
                p1->next = curr;
                p1 = p1->next;
            }
            else {
                p2->next = curr;
                p2 = p2->next;
            }
            curr = curr->next;
        }
        p2->next = nullptr;
        p1->next = dummyHead2->next;

        return dummyHead1->next;
    }
};