Merge k Sorted Lists ,优先级队列（堆）的使用

最新推荐文章于 2022-06-19 20:59:33 发布

angelewings

最新推荐文章于 2022-06-19 20:59:33 发布

阅读量646

点赞数

本文链接：https://blog.csdn.net/angelewings/article/details/41787777

版权

LeetCode 专栏收录该内容

19 篇文章 0 订阅

订阅专栏

struct cmp{
bool operator() (ListNode *p,ListNode *q){
return p->val>q->val; //> 小根堆
}
};

ListNode *mergeKLists(vector<ListNode *> &lists) {
priority_queue<ListNode*,vector<ListNode*>,cmp> heap;
ListNode *dummy=new ListNode(0),*tail=dummy;

for(int i=0;i<lists.size();i++)
if(lists[i])
heap.push(lists[i]);

while(!heap.empty()){
ListNode *q=heap.top();
heap.pop();
if(q->next)
heap.push(q->next);

tail->next=q;
tail=tail->next;
}
return dummy->next;
}