优先级队列(堆)合并K个升序链表

在做23合并K个升序链表这题时,发现使用优先级队列(堆)能够高效地解决此问题。

思路:直接无脑地将所有节点扔进优先级队列中(小根堆),然后依次取出所有节点,按照头插法构建结果链即可。

注意:这里C++优先级队列对于自定义数据类型的比大小的实现,是通过一个结构体内实现()运算符重载来实现的。

/**
 * 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) {}
 * };
 */


struct cmp{
    bool operator()(ListNode*&p1,ListNode*&p2){
        return p1->val<p2->val;
    }
};

class Solution {
private:
    priority_queue<ListNode*,vector<ListNode*>,cmp>que;	 	//传入自定义比较器cmp
public:
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        //将所有节点指针,放入大根堆中
        for(int i=0;i<lists.size();i++){
            ListNode*p=lists[i];
            while(p){
                que.push(p);
                p=p->next;
            }
        }
        ListNode*res=new ListNode(0);
        while(!que.empty()){
            //头插法
            ListNode*t=que.top();
            que.pop();
            t->next=res->next;
            res->next=t;
        }
        return res->next;
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值