[LeetCode] 86. Partition List

1. 原題連接:https://leetcode.com/problems/partition-list/

2. 解題思路

  1. 從頭至尾的遍歷自然維護着節點之間的相對位置
  2. 採用兩個鏈表分別記錄小於x和大於等於x的節點

3. 算法

  1. left_dummy和right_dummy分別表明兩個鏈表的啞節點
  2. 從頭開始遍歷鏈表,當節點val小於x時,更新left_dummy鏈表;不然,更新right_dummy鏈表
  3. 循環完成後,將兩個鏈表鏈接起來

4. 實現

struct ListNode {
    int val;
    ListNode *next;
    ListNode(int x) : val(x), next(NULL) {}
};
class Solution {
public:
    ListNode* partition(ListNode* head, int x) {
        ListNode left_dummy(-1);
        ListNode right_dummy(-1);
        
        ListNode *left_cur = &left_dummy;
        ListNode *right_cur = &right_dummy;
        for(ListNode *cur = head; cur != NULL; cur = cur->next){
            if(cur->val < x){
                left_cur->next = cur;
                left_cur = cur;
            }else{
                right_cur->next = cur;
                right_cur = cur;
            }
        }
        left_cur->next = right_dummy.next;
        right_cur->next = NULL;
        return left_dummy.next;
    }
};
相關文章
相關標籤/搜索