Merge K Sorted Lists -- LeetCode

Merge k sorted linked lists and return it as one sorted list. Analyze and describe its complexity.

思路：构建一个大小为K的小顶堆。每次从堆顶取走一个元素后，都将这个元素所属的链表中下一个元素加入堆中。若没有下一个元素则不添加。该过程一直持续到堆空位置。时间复杂度O(nlogk)。

用优先队列实现：

用优先队列实现的声明如下：

priority_queue<元素类型, 该类型的容器(如vector<元素类型>), 比较类(自己定义)> 名称;

比如这个题中，我的声明语句为priority_queue<ListNode *, vector<ListNode *>, comp> q;

这里，ListNode指针类型的星号要与前面的类型名称用空格隔开，否则会报错。

比较类定义见具体代码。

struct comp
{
    bool operator()(ListNode* n1, ListNode* n2)
    {
        return n1->val > n2->val;
    }
};
class Solution {
public:
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        priority_queue<ListNode *, vector<ListNode *>, comp> q;
        for (auto i : lists)
            if (i) q.push(i);
        if (q.empty()) return NULL;
        ListNode* res = q.top();
        ListNode* pre = res;
        q.pop();
        if (pre->next) q.push(pre->next);
        while (!q.empty())
        {
            pre->next = q.top();
            q.pop();
            pre = pre->next;
            if (pre->next)
                q.push(pre->next);
        }
        return res;
    }
};