2021-6-30 23. 合并K个升序链表(顺序合并,优先队列)

注:

题目:
给你一个链表数组,每个链表都已经按升序排列。
请你将所有链表合并到一个升序链表中,返回合并后的链表。

示例 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 = [[]]
输出:[]

提示:
k == lists.length
0 <= k <= 10^4
0 <= lists[i].length <= 500
-10^4 <= lists[i][j] <= 10^4
lists[i] 按 升序 排列
lists[i].length 的总和不超过 10^4

题解:
方法一:顺序合并

思路
我们可以想到一种最朴素的方法:用一个变量 ans 来维护以及合并的链表,第 ii 次循环把第 ii 个链表和 ans 合并,答案保存到 ans 中。

复杂度分析
时间复杂度O(k^2 n)
空间复杂度O(1):没有用到与 k 和 n 规模相关的辅助空间,故渐进空间复杂度为 O(1)。

/**
 * 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* mergetwolists(ListNode* l1,ListNode* l2){
        if(l1==nullptr){
            return l2;
        } 
        else if(l2==nullptr){
            return l1;
        }
        else if(l1->val>l2->val){
            l2->next=mergetwolists(l1,l2->next);
            return l2;
        }
        else {
            l1->next=mergetwolists(l1->next,l2);
            return l1;
        }
        return nullptr;
    }
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        if(lists.size()==0){
            return nullptr;
        }
        ListNode* res=lists[0];
        for(int i=1;i<lists.size();i++){
            res=mergetwolists(res,lists[i]);
        }
        return res;
    }
};

方法二:使用优先队列合并

思路
这个方法和前两种方法的思路有所不同,我们需要维护当前每个链表没有被合并的元素的最前面一个,k 个链表就最多有 k 个满足这样条件的元素,每次在这些元素里面选取 val 属性最小的元素合并到答案中。在选取最小元素的时候,我们可以用优先队列来优化这个过程。

复杂度分析
时间复杂度:考虑优先队列中的元素不超过 k 个,那么插入和删除的时间代价为 O(logk),这里最多有 kn 个点,对于每个点都被插入删除各一次,故总的时间代价即渐进时间复杂度为 O(kn×logk)。
空间复杂度:这里用了优先队列,优先队列中的元素不超过 k 个,故渐进空间复杂度为 O(k)。

class Solution {
public:
    class node{
    public:
        int val;
        ListNode* ptr;
        node(int _val,ListNode* _ptr){
            val=_val;
            ptr=_ptr;
        }
        bool operator <(const node &rhs) const{
            return val>rhs.val;
        }
    };
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        priority_queue<node> q;
        for(int i=0;i<lists.size();i++){
            if(lists[i]!=nullptr){
                node temp(lists[i]->val,lists[i]);
                q.push(temp);
            }
        }
        ListNode* dummyhead=new ListNode();
        ListNode* head=dummyhead;
        while(!q.empty()){
            node temp=q.top();
            head->next=temp.ptr;
            head=head->next;
            q.pop();
            if(temp.ptr->next!=nullptr){
                node x(temp.ptr->next->val,temp.ptr->next);
                q.push(x);
            }
        }
        return dummyhead->next; 
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值