23. 合并K个元素的有序链表

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

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

示例:

输入:
[
  1->4->5,
  1->3->4,
  2->6
]
输出: 1->1->2->3->4->4->5->6

思路:每两个链表一合并,每一次规模缩减为原来的一半,直到最后剩两个链表合并为一个链表。

复杂度分析: k为链表的个数,n为每个链表的长度。第一次两两合并是进行了k/2次,每次处理2n个值。第二次两两合并是进行了k/4次,每次处理4n个值。。。。最后(设第x次)一次两两合并,进行了一次合并(k/(2^x)=1, 因为最后一组是一个链表,求出x=logk),每次处理 n*k个值。

所以时间复杂度为O(nklogk),因为每个链表长度可能不一样,nk就是所有链表总长度,因此O(n)=O(Nlogk)(N为k个链表总长度)

class Solution {
public:
    ListNode* merge(ListNode* l1, ListNode* l2) {
        ListNode *dummy=new ListNode (0), *p=dummy;
        while(l1 && l2){
            if(l1->val >= l2->val){
                p->next=l2;
                l2=l2->next;
            }
            else{
                p->next=l1;              
                l1=l1->next;
            }
            p=p->next;
        }
        if(l1) p->next=l1;
        if(l2) p->next=l2;
        return dummy->next;
    }
 
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        if(lists.empty()) return nullptr;
        while(lists.size()>1){
            vector<ListNode*>temp;
            for(int i=1; i<lists.size(); i=i+2){
                temp.push_back(merge(lists[i-1], lists[i]));
            }
            if(lists.size()%2==1){
                temp.push_back(lists.back());
            }
            lists=temp;
        }
        return lists[0];
    }
 
    // 合并过程这样做也可以
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        if(lists.empty()) return nullptr;
        queue<ListNode*>q(lists.begin(), lists.end());
        while(q.size()>1){
            ListNode *a=q.front();
            q.pop();
            ListNode *b=q.front();
            q.pop();
            q.push(merge(a,b));
        }
        return q.front();
    }
};

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值