23. Merge k Sorted Lists - hard

这道题可以利用merge 2 sorted list。想到D&C, K个list可以被二分,分到最后只有2个list的情况。
画成树的话,一共有lgK个level, mergeTwoList的时间是O(n),
所以这道题利用Divide and Conquer,和merge2list,running time应该是O(nlgK):

看见有人用priority queue和comparator做。但速度并没有提升。
下次刷的时候,研究研究那种做法。

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
public class Solution {
    public ListNode mergeKLists(ListNode[] lists) {
        int len = lists.length;
        if (lists == null || len == 0)  return null;
        else if (len == 1)  return lists[0];
        else {
            ListNode[] list1 = Arrays.copyOfRange(lists, 0, len/2);
            ListNode[] list2 = Arrays.copyOfRange(lists, len/2, len);
            ListNode l1 = mergeKLists(list1);
            ListNode l2 = mergeKLists(list2);
            ListNode l3 = mergeTwoLists(l1, l2);
            return l3;
        }
    }
    
    private ListNode mergeTwoLists(ListNode l1, ListNode l2) {
        if (l1 == null) return l2;
        if (l2 == null) return l1;
        if (l1.val < l2.val) {
            l1.next = mergeTwoLists(l1.next, l2);
            return l1;
        }
        else {
            l2.next = mergeTwoLists(l1, l2.next);
            return l2;
        }
    }
}

你可能感兴趣的:(23. Merge k Sorted Lists - hard)