【每日一题】23. 合并K个排序链表

题目

合并 k 个排序链表，返回合并后的排序链表。请分析和描述算法的复杂度。

示例

输入
[
1->4->5,
1->3->4,
2->6
]
输出
1->1->2->3->4->4->5->6

思路

参考力扣大佬的思路，仅仅将比较器使用lambda表达式将语法简化。
主要思路是使用优先队列，步骤如下：
1.依次将各个有序链表的表头压入优先队列（小顶堆）；
2.构建一个虚拟表头，当队列非空时，将队列中的元素弹出，接在虚拟表头后面；
3.如果弹出的节点后续节点非空，则将后续节点压入队列；
4.直到队列为空，返回虚拟节点的后续节点。

代码

class Solution {
    public ListNode mergeKLists(ListNode[] lists) {
        if (lists == null || lists.length == 0) {
            return null;
        }
        PriorityQueue<ListNode> queue = new PriorityQueue<> (lists.length, (node1, node2) -> node1.val - node2.val);
        ListNode dummy = new ListNode(0);
        ListNode p = dummy;
        for (ListNode node : lists) {
            if (node != null) {
                queue.add(node);
            }
        }
        while (!queue.isEmpty()) {
            p.next = queue.poll();
            p = p.next;
            if(p.next != null) {
                queue.add(p.next);
            }
        }
        return dummy.next;
    }
}