[leetcode]Sort List

鏈表用歸併排序，以前也作過，用遞歸寫很簡單，代碼不長也很好理解。地址

可是今天無聊想用非遞歸寫下。

基本思想就是每次就是1,2,4,8,16這樣分塊。。而後比較，合併。

想法很簡單，結果代碼寫出來也debug了一小會兒。

主要遇到的問題就是，在merge兩個list的時候(其實就是一個的兩部分)，怎麼處理末尾和開始。

沒處理好，可能就是個環。

仍是在代碼裏面注視說好點。

 

class Solution {
public:
    ListNode *get_next(ListNode* head, int n) {
        while(n-->0 && head) {
            head = head->next;
        }
        return head;
    }
    
    ListNode* merge2(ListNode* l1, ListNode* l2, int n, int pos, int len) {
        ListNode* dummy = new ListNode(-1);
        ListNode* curr = dummy;
        ListNode* le = l2;
        int n1 = min(n, len - pos); // 計算l1有多少個節點
        int n2 = min(n, len - (pos+n)); // 計算l2有多少個節點
        while(n1 > 0 && n2 > 0) { //開始是用l1,l2這樣表示的，明顯的問題，只是其中一段，next確定不是nullptr
            if (l1->val < l2->val) {
                curr->next = l1;
                l1 = l1->next;
                n1--;
            } else {
                curr->next = l2;
                n2--;
                l2 = l2->next;
            }
            curr = curr->next;
        }
        if (n1 > 0&& l1) curr->next = l1;
        if (n2 > 0&& l2) curr->next = l2;
        ListNode* ans = dummy->next;
        delete dummy;
        return ans;
    }
    
    ListNode *sortList(ListNode *head) {
        if (head == nullptr || head->next == nullptr) return head;
        int len = 0;
        for (ListNode* tmp = head; tmp; tmp = tmp->next) {
            len++;
        }
        int step = 1; // step = 1,2,4,6,8

        while (step < len ) {
            ListNode* curr = head;
            ListNode* prev = nullptr;
            for (int i = 0; i < len; i += step*2) { //i其實就是curr在list裏面的位置
                ListNode* next = get_next(curr, step*2); //記錄next...否則前面重排以後很差找
                ListNode* tmp = merge2(curr, get_next(curr, step), step, i, len); // l1 是當前節點的step個，l2是step+1到step*2個
                if (i == 0) head = tmp; //每次迭代head可能都會變
                curr = tmp;
                if (prev) prev->next = curr; //把前面的指向如今的head
                prev = get_next(curr, step * 2 - 1);  //這個是如今l1,l2合併後的最後一個節點，就是後面的prev
                curr = next; //curr指向下一個
            }
            step *= 2;
        }
        ListNode* curr = get_next(head, len - 1);//最後的節點的next其實可能還指向了某些地方的 2->1這種，咱們排序以後是1->2可是2仍是指向1的，就有環
        curr->next = nullptr;
        return head;
    }
    
};