[LeetCode 23] Merge k Sorted Lists (最小堆)

23. Merge k Sorted Lists

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

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */

---

题解：
此题为二路归并的推广。回想二路归并的做法，是两个已经有序的序列，然后不断的取出其中的最小值，然后将其加入到新的序列，因为两序列本身有序，所以找到下一个最小值只需要将被取出最小值的序列的下标加1，然后再比较。每次比较是O(1)的，因此总体复杂度为O(n)。
推广到k路归并，朴素的算法是每次遍历序列头，找到其中最小者加入新序列，然后将被取出者的下标加1。这样，每次找出最小值的复杂度为O(k)，因此总复杂度为O(nk)。
上述算法有明显的可优化空间，我们有堆(Heap)这样的数据结构来快速找到最小值，因此，只需在上述思路的基础上，将遍历查找改为使用堆来查找，就可将单次操作的复杂度优化到O(logk)，总复杂度降到O(nlogk)。
还要注意此题要归并的是链表，因此在处理上与处理数组有所不同。

代码：

#include <queue>
#include <functional>

//ListNode*的比较器，用于ListNode*的优先队列
class Compare {
public:
    bool operator()(ListNode* p1, ListNode* p2) {
        return p1->val > p2->val;
    }
};

class Solution {
public:
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        priority_queue<ListNode*, vector<ListNode*>, Compare > que;

        int k = lists.size();
        for (int i = 0; i < k; ++i) {
            if (lists[i] != NULL)    //排除空表的情况
                que.push(lists[i]);
        }

        //新建一个头结点，方便处理
        ListNode* pHead = new ListNode(-1);
        ListNode* pCur = pHead;

        while (!que.empty()) {
        //取最小
            ListNode *tmp = que.top();
            que.pop();

        //加入新序列
            pCur->next = tmp;
            pCur = pCur->next;

            //新元素入堆
            if (tmp->next != NULL) {
                que.push(tmp->next);
            }
        }
        pCur = pHead->next;
        delete pHead;     //删除头结点

        return pCur;
    }
};