LeetCode23. 合并K个升序链表

题目

分析

本题思想不难，就是仿照合并两个升序链表的思路，假设 k 个 链表，最长的链表长度为 n，那么只用将 k 个链表表头元素相比找到最小的

插到新的链表表尾即可。这样的话时间复杂度 为 O（n * k）。我们可以进一步优化，k个链表表头元素比较取最小， 我们可以采用堆，这样

取最小的时间为O（1），然后再把链表下一个元素插到堆里面的时间复杂度为O（log K），总体时间复杂度就被优化为O（n * logk）

 

有关priority_queue来实现小根堆的链接 ：https://www.cnblogs.com/huashanqingzhu/p/11040390.html

代码

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:

    struct Cmp{
        bool operator() (ListNode *a,ListNode *b){
            return a->val > b->val;
        }
    };
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        priority_queue<ListNode*,vector<ListNode*>,Cmp>heap;
        auto dummy = new ListNode(-1),cur = dummy;

        for(auto l :　lists){
            if(l) heap.push(l);
        }

        while(heap.size()){
            auto t = heap.top();
            heap.pop();

            cur->next = t;
            cur = cur->next;

            if(t->next) heap.push(t->next);
        }

        return dummy->next;

        
    }
};