Leetcode Merge k Sorted Lists

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

vinllen

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

阅读量658

点赞数

CC 4.0 BY-SA版权

分类专栏： Leetcode 文章标签： leetcode

本文链接：https://blog.youkuaiyun.com/u011353822/article/details/40189395

Leetcode 专栏收录该内容

14 篇文章

订阅专栏

本文介绍了一种使用小顶堆解决合并K个已排序链表问题的方法，并提供了详细的C++实现代码。通过维护一个小顶堆，每次弹出最小元素并将其后续元素加入堆中，最终合并所有链表为一个有序链表。

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

题目：

Merge k sorted linked lists and return it as one sorted list. Analyze and describe its complexity.

解题：

维持一个小顶堆，每次弹出最小的元素，弹出之后把该元素在原数组中的下一个元素压栈（如果不为空）。

代码：

class Solution {
public:
    ListNode *mergeKLists(vector<ListNode *> &lists) {
        priority_queue<Node> q;
        ListNode *head = new ListNode(-1);
        ListNode *pre = head;
        for(int i = 0; i < lists.size(); i ++) {
            if(lists[i] != NULL) {
                Node now;
                now.x = lists[i];
                q.push(now);
            }
        }
        while(!q.empty()) {
            Node top = q.top();
            q.pop();
            pre->next = top.x;
            if(top.x->next) {
                Node now;
                now.x = top.x->next;
                q.push(now);
            }
            pre = top.x;
        }
        return head->next;
    }
private:
    struct Node {
        ListNode *x;
        bool operator<(const Node p) const{
            return x->val > p.x->val;
        }
    };
};