leetcode 23. Merge k Sorted Lists

写在前面

又是一道面试题，类似的，都是有很容易想到的AC解，但是并不是最优解，本题很好地利用了优先队列的性质（类似文件排序中的最小堆用法）。

题目描述

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

思路分析

最笨的解法无疑是两两合并list，直到只剩下一个list为止，这种解法，无疑会访问元素多次，并非最优解。如果使用优先队列，首先保存所有的最小元素，最小堆顶端的元素一定是List的开始元素，对于次小元素，一定是当前最小元素的下一个元素或者是堆中剩余元素的最小元素（利用已排序的性质，当前最小元素的下一个元素比当前最小元素大，但可能比堆中所有元素小，因此可以最小堆元素的候选，不可能是其它list中的剩余元素，因为它们已经比堆中元素大了），我们把这个元素加入最小堆作为候选，递归直到堆为空，根据这种思路，我们写出如下代码（需要重新定义堆的比较函数）：

/** * Definition for singly-linked list. * struct ListNode { *     int val; *     ListNode *next; *     ListNode(int x) : val(x), next(NULL) {} * }; */struct Cmp {    bool operator()(const ListNode* lhs,const ListNode* rhs) {        return lhs->val>rhs->val;    }};class Solution {public:    ListNode* mergeKLists(vector<ListNode*>& lists) {        if(lists.size()==0) return nullptr;        priority_queue<ListNode*,vector<ListNode*>,Cmp> mQueue;        for(auto val: lists) {            if(val)mQueue.push(val);        }        ListNode* dummy = new ListNode(0);        auto head = dummy;        while(!mQueue.empty()) {            head->next = mQueue.top();            mQueue.pop();            head = head->next;            if(head->next) mQueue.push(head->next);        }        head = dummy->next;        delete dummy;        dummy = nullptr;        return head;    }};

需要注意对最小堆的定义。