思路:看到排序,我就很自然的想到最大最小堆,本题适用于最小堆的优先队列,依次从队列中取出队顶元素即可,当然也可以像合并两条链表(力扣21题刷题笔记----每日一题:力扣21. 合并两个有序链表)那样做,不过复杂度比较大。
/**
* 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:
ListNode* mergeKLists(vector<ListNode*>& lists) {
ListNode *l = new ListNode(-1),*p = l;//p为虚拟头结点
priority_queue<ListNode*,vector<ListNode*>,cmp> pq; //构建最小堆
//判断是否为空
if(lists.size() == 0)
return nullptr;
for(int i=0;i<lists.size();++i){
//判断链表是否为空,不为空,则将头结点加入优先队列
if(lists[i])
pq.push(lists[i]);
}
while(!pq.empty()){
//队列不为空,将队顶元素取出
ListNode *node = pq.top();
pq.pop();
//更新l节点
l->next = node;
l = l->next;
//若后继节点不为空,则将其加入到队列中
if(node->next !=nullptr )
pq.push(node->next);
}
return p->next;
}
struct cmp{
bool operator()(ListNode *p,ListNode *q){
return p->val > q->val; //构建最小堆
}
};
};