23. 合并 K 个升序链表

  1. 题目链接

  2. 解题思路:每次从全局的拿一个最小值出来,每个链表的「头」,都是最小的,所以,我们可以使用一个小根堆(优先级队列),存放每个链表当前的「头」,然后弹出一个全局最小的节点出来,然后把该节点的next放回小根堆,供之后使用。

    • 注意,压入小根堆时,要保证不为nullptr。
  3. 代码

    /**
    * 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 MyCompare{
    bool operator()(ListNode *node1, ListNode *node2) {
    return node1->val > node2->val;
    }
    };
    ListNode* mergeKLists(vector<ListNode*>& lists) {
    priority_queue<ListNode*, vector<ListNode*>, MyCompare> que; // 小根堆
    for (int i = 0; i < lists.size(); ++i) {
    if (lists[i] != nullptr) {
    que.push(lists[i]);
    }
    }
    if (que.empty()) {
    return nullptr;
    }
    ListNode *ans = que.top();
    ListNode *cur = ans;
    que.pop();
    if (ans->next != nullptr) {
    que.push(ans->next);
    }
    while(!que.empty()) {
    cur->next = que.top();
    que.pop();
    cur = cur->next;
    if (cur->next != nullptr) {
    que.push(cur->next);
    }
    }
    return ans;
    }
    };
posted @   ouyangxx  阅读(11)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 终于写完轮子一部分:tcp代理 了,记录一下
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· 别再用vector<bool>了!Google高级工程师:这可能是STL最大的设计失误
· 单元测试从入门到精通
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理
点击右上角即可分享
微信分享提示