合并k个有序链表

leetcode题目：

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

如果每次遍历k个链表寻找最小值，那么时间复杂度为O(n*k)，超时。

用堆可以将时间复杂度降低到O(n*lgk)。

申请一个大小为K的堆，堆里的存放每个链表表头的元素，每次从堆中取出最小值，并将取出来的最小值的后一个节点加入到堆中，如果是null就不用加入了。

持续上述过程，直到堆里一个元素都没有了。

细节：PriorityQueue初始化构造函数要指定比较器时，必须要指定初始容量，以前不记得这样用。

代码如下：

   public boolean valid(ListNode[] lists){

        if(lists==null || lists.length==0){ return false; }
        
        int count = 0;
        for(int i=0;i<lists.length;i++){
            if(lists[i]!=null){
                count++;
            }
        }
        if(count==0){return false;}
        return true;
    }

    //合并k个有序链表
    public ListNode mergeKLists(ListNode[] lists) {

        if(!valid(lists)){
            return null;
        }

        ListNode head = null;

        //用PriorityQueue来实现一个堆，同时指定比较器
        PriorityQueue<ListNode> heap = new PriorityQueue<ListNode>(1,new Comparator<ListNode>(){
            @Override
            public int compare(ListNode o1, ListNode o2) {
                return o1.val-o2.val;
            }
        });

        for(int i=0;i<lists.length;i++){    //将所有链表头结点加入到堆中
            if(lists[i]!=null){
                heap.add(lists[i]);
            }
        }

        ListNode min = heap.poll();         //寻找头结点
        head = min;
        min = min.next;

        if(min!=null)
            heap.add(min);
        ListNode last = head;              //ListNode保存已经排好序的链表的最后一个节点

        while (!heap.isEmpty()){
            ListNode minNode = heap.poll();
            last.next = minNode;         //更新last
            last = minNode;
            minNode = minNode.next;
            if(minNode!=null){
                heap.add(minNode);
            }
        }

        return head;
    }