Leetcode.23. 合并 K 个升序链表

最新推荐文章于 2024-10-10 20:14:35 发布

圣昔

最新推荐文章于 2024-10-10 20:14:35 发布

阅读量155

点赞数 1

文章标签：链表数据结构算法 leetcode c++

本文链接：https://blog.csdn.net/Xiaoyue_liang/article/details/131724927

版权

题目大意

给你一个链表数组，每个链表都已经按升序排列。

请你将所有链表合并到一个升序链表中，返回合并后的链表。

示例一

输入：lists = [[1,4,5],[1,3,4],[2,6]]
输出：[1,1,2,3,4,4,5,6]
解释：链表数组如下：
[
  1->4->5,
  1->3->4,
  2->6
]
将它们合并到一个有序链表中得到。
1->1->2->3->4->4->5->6

示例二

输入：lists = []
输出：[]

示例三

输入：lists = [[]]
输出：[]

解题思路

用一个小根堆来存储这k个结点，每次弹出堆顶结点，并把下一个结点插入堆中。

实现代码及注释

/**
 * 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:
    struct cmp{
        bool operator()(ListNode *a,ListNode *b)//自定义一个比较函数，用来找到最小的那个结点
        {
            return a->val>b->val;
        }
    };
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        priority_queue<ListNode*,vector<ListNode*>,cmp> que;//建立一个小根堆，用来储存链表的结点
        for(int i=0;i<lists.size();i++){//把每个链表的头结点加入到que中
            if(lists[i]) que.push(lists[i]);
        }
        ListNode dummy(-1);//定义虚拟头结点
        ListNode* p=&dummy;//定义一个指针指向当前虚拟头结点
        while(!que.empty())//当堆不为空时
        {
            ListNode* top=que.top();
                que.pop();//弹出堆顶元素
            p->next=top;//把结点插入到指针的下一个结点
            p=top;//当前指针后移
            if(top->next)//如果被弹出的结点存在下一个结点时
                que.push(top->next);//就将其插入到下一个结点位置
        }
        return dummy.next;//最后返回出
    }
};