23. Merge k Sorted Lists (Divide and conquer, Linked List) 以及java匿名内部类

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

Example:

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

分析;

1. 分而治之。也就是把问题分解成两两合并直到全部合并成一个为止。时间复杂度O((m/2)(k/2 + k/4 + k/8 + …)logk),还是O(kmlogk),空间复杂度为O(1).

合并的时候可以像下图一样i和i+interval合并,也可以i和length-i合并。

分析i和i+interval合并: interval每次乘以2, 在一个循环内,两两合并的是lists[i]和lists[i+index](注意判断越界); 更新i=i+2*interval (这里一开始写成了i=i+interval+1然后就runtime error了。。)

两个链表合并的问题见Leetcode21,哪个链表当前元素小就将cur只想这个节点,并更新这个节点为下一个,需要注意用一个start保存一个初始节点,然后返回的是start.next.(这样可以避免为空的情况),还有每次cur = cur.next指向下一个。

Divide_and_Conquer

 public ListNode mergeKLists(ListNode[] lists) {
        if(lists==null || lists.length==0)
            return null;
        int len = lists.length;
        int interval = 1;
        while(interval<len){
            for(int i=0;i<len && (i+interval)<len;){
                lists[i] = merge2(lists[i], lists[i+interval]);
                i = i+ interval*2;//注意i的更新规则
            }
            interval *= 2;
        }
        return lists[0];
    }
    private ListNode merge2(ListNode l1, ListNode l2){//合并两个链表
        if(l1==null || l2==null){
            if(l1==null)
                return l2;
            else 
                return l1;
        }
        ListNode start = new ListNode(0);
        ListNode cur = start;

        while(l1!=null && l2!=null){
            if(l1.val>l2.val){
                cur.next = l2;
                l2 = l2.next;
            }
            else{
                cur.next = l1;
                l1 = l1.next;
            }
            cur = cur.next;
        }
        if(l1!=null)
            cur.next = l1;
        if(l2!=null)
            cur.next = l2;
        return start.next;
    }

2. 使用PriorityQueue

用优先队列,每次插入元素都进行排序比较,最前面的元素小,所以需要new一个Comparator. 先把k个list的第一个元素放进heap,每次poll出一个后,放进poll出的那个的后一个,直到heap为空。时间复杂度O(mklogk),空间复杂度为O(k).

这里就需要清楚PriorityQueue的用法以及怎么使用匿名内部类:

http://www.importnew.com/6932.html (PriorityQueue)

https://www.journaldev.com/996/java-inner-class (Java匿名类 还有内部类其他两种形式的介绍)

public ListNode mergeKLists(ListNode[] lists) {
        if (lists == null || lists.length == 0) return null;
        if (lists.length == 1) return lists[0];
        PriorityQueue<ListNode> queue = new PriorityQueue<>(new Comparator<ListNode>() {
            public int compare(ListNode o1, ListNode o2) {
                return o1.val - o2.val;
            }
        });
        for (int i = 0; i < lists.length; i++) {
            if (lists[i] != null) queue.add(lists[i]);
        }
        ListNode dummy = new ListNode(-1), cur = dummy;
        while (!queue.isEmpty()) {
            ListNode temp = queue.poll();
            cur.next = temp;
            cur = temp;
            if (temp.next != null) queue.add(temp.next);
        }
        return dummy.next;
    }

参考https://blog.csdn.net/katrina95/article/details/79112038 

Heap与PriorityQueue的区别: https://blog.csdn.net/roufoo/article/details/80638476

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值