【LeetCode】23.K个有序链表合并

最新推荐文章于 2024-02-20 21:30:06 发布

Freeman_zxp

最新推荐文章于 2024-02-20 21:30:06 发布

阅读量328

点赞数

分类专栏：基础算法文章标签： leetcode c++

本文链接：https://blog.csdn.net/zpalyq110/article/details/89242056

版权

基础算法专栏收录该内容

72 篇文章 0 订阅

订阅专栏

23. Merge k Sorted Lists

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

Example:

Input:
[
  1->4->5,
  1->3->4,
  2->6
]
Output: 1->1->2->3->4->4->5->6

方法1：堆

Time complexity : $O\left ( n*log(k)\right )$
Space complexity : $O\left ( n \right )$
思路：
我们首先把k个链表的首元素都加入最小堆中，它们会自动排好序。然后我们每次取出最小的那个元素加入我们最终结果的链表中，然后把取出元素的下一个元素再加入堆中，下次仍从堆中取出最小的元素做相同的操作，以此类推，直到堆中没有元素了，此时k个链表也合并为了一个链表，返回首节点即可，

struct compare{
    bool operator()(ListNode* a, ListNode* b){        
    return a->val > b->val;
    }
};

class Solution {
public:
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        priority_queue<ListNode*, vector<ListNode*>, compare> q;
        for(auto node : lists){
            if(node) q.push(node);
        }
        ListNode *res = new ListNode(-1);
        ListNode *cur = res;
        while(!q.empty()){
            auto top = q.top();
            q.pop();
            cur->next = top;
            cur = cur->next;
            if(cur->next)
                q.push(cur->next);
                
        }
        return res->next;
    }
};