合并K个排序链表

题目链接:合并K个排序链表
题目描述:
合并 k 个排序链表,返回合并后的排序链表。请分析和描述算法的复杂度。

示例:

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

题目分析:基于合并2个有序链表的过程,合并K个链表,最简单的方式就是,将K个链表进行K-1次合并(每次合并2个,并保留上一次合并的结果进行下一次合并),这个时间复杂度是O(n2),寻求更低的时间复杂度是我们不断追求的目标。合并K个链表的过程是不是觉得似曾相识呢?没错,就是归并排序,这个基本上就是归并排序了,每次进行两两归并,归并的过程就是组合链表的过程,时间复杂度是O(nlogn)
代码:

class Solution {

    public ListNode mergeKLists(ListNode[] lists) {
        return mergeLists(lists,0,lists.length-1);
    }
    public ListNode mergeLists(ListNode[] st,int l,int r){
        //分组只有一个元素了,开始归并
        if(r-l==0) return st[l];
        if(l>r) return null;
        int m = l+(r-l)/2;
        ListNode a,b,head = new ListNode(-1);
        //区间为左闭右闭
        a = mergeLists(st,l,m);
        b = mergeLists(st,m+1,r);
        ListNode p = head;
        //
        while(a!=null&&b!=null){
            if(a.val<b.val){
                p.next = a;
                a = a.next;
            }else{
                p.next = b;
                b = b.next;
            }
            p = p.next;
        }
        p.next = a==null?b:a;
        return head.next;
    }
}

继续思考:仔细观察一下合并的过程,可以发现,每次都是选取的K个链表中第一个元素的最小值作为新链表的节点,那么可以使用一个大小为K的优先队列来维护这些值,新链表的值只需要不断从队列取就可以了

class Solution {

public ListNode mergeKLists(ListNode[] lists) {
        PriorityQueue<ListNode> q = new PriorityQueue<>(Comparator.comparingInt(o -> o.val));
        for (ListNode head: lists) {
            if(head!=null) q.add(head);
        }
        ListNode head = new ListNode(-1);
        ListNode p = head;
        while(!q.isEmpty()){
          //取最小的
           ListNode te = q.poll();
           p.next = te;
           p = p.next;
           if(te.next!=null) q.add(te.next);//继续加入新的第一个节点
        }
        return head.next;
    }
}

不过这个执行时间好像更长了耶。。。,体会思路,思路哈😁

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值