合并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;
}