23. Merge k Sorted Lists 合并K个有序链表

最新推荐文章于 2024-04-03 21:24:25 发布

知之可否

最新推荐文章于 2024-04-03 21:24:25 发布

阅读量547

点赞数

分类专栏： leetcode

本文链接：https://blog.csdn.net/gao1440156051/article/details/51757848

版权

leetcode 专栏收录该内容

164 篇文章 1 订阅

订阅专栏

https://leetcode.com/problems/merge-k-sorted-lists/

注意一点：小顶堆要重载>符号

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    struct cmp{
        bool operator()(const ListNode* a,const ListNode* b)const{
            return a->val>b->val;
        }
    };
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        priority_queue<ListNode*,vector<ListNode*>,cmp> pq;
        int len=lists.size();
        for(int i=0;i<len;i++){
            if(lists[i]){
                pq.push(lists[i]);
            }
        }

        ListNode dummy(0);
        ListNode* p=&dummy;
        while(!pq.empty()){
            ListNode* top=pq.top();
            pq.pop();
            p->next=top;
            p=p->next;

            if(top->next){
                pq.push(top->next);
            }
        }

        return dummy.next;
    }
};