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;
}
};