LeetCode OJ-23-Merge k Sorted Lists

题目：

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

大意：

合并k个排好的的单链表。分析和描述它的复杂性。

思路：

使用小顶堆来实现，先将K个链表的头结点入堆，取堆顶元素，这个结点就是最小的，接着让取出的这个结点的下一个结点入堆，再取堆顶元素，其为第二小的结点，一直这样子操作，直到所有的结点都处理完，这样就完成了链表的合并，更多细节请见代码和注释。假设K个链表一共有N个结点，时间复杂度是O(N*log(K))，空间复杂度是O(K)。
在本题中，直接使用优先级队（PriorityQueue）列实现小顶堆，较方便。
有关优先级队列的相关知识，可以参考以下几篇博客：
http://blog.csdn.net/a15994269853/article/details/20571317
http://blog.csdn.net/zhouguosen_always/article/details/16823595
http://www.jb51.net/article/84371.htm
http://cuisuqiang.iteye.com/blog/2019157

代码：

/**
 * 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) {
        if(lists == null || lists.length == 0) {
            return null;
        }

        Queue<ListNode> pq = new PriorityQueue<>(new Comparator<ListNode>() {
            //重写compare()方法
            public int compare(ListNode l1, ListNode l2) {
                return l1.val - l2.val;
            }
        });

        //优先级队列中不允许使用null元素，要判断是否为非空
        for(ListNode list: lists) {
            if(list != null) {
                pq.add(list);
            }
        }

        ListNode head = new ListNode(0);
        ListNode p = head;

        //注意!pq.isEmpty()与pq != null是不一样的
        //pq!=null是判断pq对象是不是为空，比如List pq =null;那么pq就是null的
        //若List pq = new ArrayList();这个时候pq.isEmpty()是true的，
        //而pq!=null是false
        //在这个问题中，每次将小顶堆的堆顶取出，最后所有元素被取出，只能用isEmpty来判断
        //pq != null一直是true的，若以此作为判断条件将无法跳出while循环从而报错
        while(!pq.isEmpty()) {
            //取出队列头元素，也就相当于小顶堆的堆顶，再将取出结点的下一个结点放入队列中
            //随后会生成新的小顶堆
            ListNode n = pq.poll();
            p.next = n;
            p = p.next;
            if(n.next != null) {
                pq.add(n.next);
            }
        }

        return head.next;
    }
}

PS
有时间阅读http://blog.csdn.net/derrantcm/article/details/46905087
大神自己实现了一个小顶堆及相关功能。