23.合并K个升序链表【HARD】

u0uu

已于 2024-08-03 15:09:46 修改

阅读量153

点赞数 1

分类专栏： leetcode算法文章标签：链表算法 javascript

于 2024-07-31 19:40:53 首次发布

本文链接：https://blog.csdn.net/u010012355/article/details/140826899

版权

leetcode算法专栏收录该内容

14 篇文章 0 订阅

订阅专栏

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

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

示例 1：

输入：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

示例 2：

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

示例 3：

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

解题：首先想最简单的，最小堆，优先队列。在优先队列里，允许数字重复，并且还有序，最适合排序。

/**
 * 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:
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        priority_queue<int, vector<int>, greater<int>> heap;
        ListNode* p;
        for (int i = 0; i < lists.size(); i++) {
            p = lists[i];
            while (p) {
                heap.push(p->val);
                p = p->next;
            }
        }
        ListNode* res = new ListNode(0);
        p = res;
        while (!heap.empty()) {
            ListNode* tmp = new ListNode(heap.top());
            p->next = tmp;
            p = tmp;
            heap.pop();
        }
        return res->next;
    }
};