[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;
    }
    
};
原文地址:https://www.cnblogs.com/x1957/p/4229987.html