力扣 hot100 Day40

23. 合并 K 个升序链表

给你一个链表数组,每个链表都已经按升序排列。

请你将所有链表合并到一个升序链表中,返回合并后的链表。

//自己写的垃圾
class Solution {
public:
    ListNode* mergeKLists(vector& lists) {
        vector record;
        int n = lists.size();
        for(int i=0;ival);
                lists[i]=lists[i]->next;
            }
        }
        if (record.empty()) {
            return nullptr;
        }
        sort(record.begin(),record.end());
        ListNode* res = new ListNode(record[0]);
        ListNode* cur = res;
        int len = record.size();
        for(int i=1;inext = new ListNode(record[i]);
            cur = cur->next;
        }
        return res;
    }
};

没有思考纯粹取巧,放数组里排序后生成新的链表,回去等通知版

//抄的
class Solution {
public:
    ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) {
        ListNode dummy(0);
        ListNode* tail = &dummy;
        
        while (l1 && l2) {
            if (l1->val < l2->val) {
                tail->next = l1;
                l1 = l1->next;
            } else {
                tail->next = l2;
                l2 = l2->next;
            }
            tail = tail->next;
        }
        
        tail->next = l1 ? l1 : l2;
        return dummy.next;
    }

    ListNode* mergeKLists(vector& lists) {
        if (lists.empty()) return nullptr;
        
        int k = lists.size();
        while (k > 1) {
            for (int i = 0; i < k / 2; ++i) {
                lists[i] = mergeTwoLists(lists[i], lists[k - 1 - i]);
            }
            k = (k + 1) / 2;
        }
        
        return lists[0];
    }
};

面试该写的算法,分治归并算法

逻辑说起来也很简单,两两合并的意思,为了方便循环,一头一尾开始合并,然后逐次减半k值。

比较需要注意的就是k减半的计算,举例子算算就好了

每层的时间复杂度都是 O(N),共有 log₂K 层。​​总时间复杂度 = O(N log K)​​。

你可能感兴趣的:(Hot100,leetcode,算法,职场和发展)