Merge k Sorted Lists

leetcode

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

使用一个大小为K的堆来维持找当前的最小元素。

一、Java当中的PriorityQueue如何使用？

class ListNode {
int val;
ListNode next;
ListNode(int x) { val = x; }
}

对于上面的节点，需要重新实现Comparator接口

实现方式为：

class MyCompare implements Comparator{
public int compare(Object o1, Object o2){
ListNode n1 = (ListNode)o1;
ListNode n2 = (ListNode)o2;
if(n1.val < v2.val){
return -1;
}else if(n1.val==n2.val){
return 0;
}else{
return 1;
}
}
}

注意上面的定义方式，对于升序排列，则当小于时返回-1，大于时返回1，else 返回 0.

另外也可以也可以使用匿名函数的方式。

Comparator<ListNode> mc = new Comparator<ListNode>(){
public int compare(ListNode n1, ListNode n2){
//should pay attention to the return value below
if(n1.val < n2.val){
return -1;
}else if(n1.val==n2.val){
return 0;
}else{
return 1;
}
}
};

算法总体的代码实现为：

public class Solution {
public ListNode mergeKLists(ListNode[] lists) {
if(lists==null||lists.length==0) return null;
Comparator<ListNode> mc = new Comparator<ListNode>(){
public int compare(ListNode n1, ListNode n2){
//should pay attention to the return value below
if(n1.val < n2.val){
return -1;
}else if(n1.val==n2.val){
return 0;
}else{
return 1;
}
}
};
　　　　 //这里需要注意，在新的java版本当中，不能指定泛型的类型，也就是下面的代码如果定义为Queue<ListNode> queue = new PriorityQueue<ListNode>(lists.length,mc);会报错
　　　　 //实际当中，应该是把其当做Object来看待的，在实际使用时，可以对其进行强制类型转化，即可

Queue queue = new PriorityQueue(lists.length,mc);
for(int i=0;i<lists.length;i++){
//there may be the contents in the lists is null, this is a corner case
if(lists[i]!=null)
queue.add(lists[i]);
}
ListNode head=null;
ListNode prenode=null;
while(!queue.isEmpty()){
ListNode top = (ListNode)queue.poll();
ListNode next =  top.next;
top.next=null;
if(head==null){
head=top;
prenode=top;
}else{
prenode.next=top;
prenode=top;
}
if(next!=null){
queue.add(next);
}
}
return head;
}
}