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

最新推荐文章于 2019-02-26 19:24:53 发布

原创最新推荐文章于 2019-02-26 19:24:53 发布 · 678 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

19 篇文章

订阅专栏

本文介绍了一种利用优先队列（堆）合并多个链表的方法，通过定义一个特殊的比较器，实现高效地合并操作。这种方法适用于链表数量较多的情况，能够减少额外的空间开销并提高合并效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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