LeetCode 23. 合并K个升序链表

LeetCode 23. 合并K个升序链表

作为第一道AC的困难难度的题目，记录一下
题目：

给你一个链表数组，每个链表都已经按升序排列。 请你将所有链表合并到一个升序链表中，返回合并后的链表。

示例 1： 输入：lists = [[1,4,5],[1,3,4],[2,6]] 输出：[1,1,2,3,4,4,5,6]
解释：链表数组如下： [ 1->4->5, 1->3->4, 2->6 ] 将它们合并到一个有序链表中得到。
1->1->2->3->4->4->5->6

第一个思路：
K路合并

第二个思路：
最小堆
将每条链表的头结点先加入到堆中，之后每一次取出堆顶元素（挂到合并后的链表中），然后将他的下一个节点加入到堆中，循环这个过程，直至优先队列为空

public ListNode mergeKLists(ListNode[] lists) {
        if(lists.length == 0)       //空链表，直接返回null
            return null;

        //创建优先队列，按照每个节点的val值升序排列
        PriorityQueue<ListNode> queue = new PriorityQueue<>(new Comparator<ListNode>() {
            @Override
            public int compare(ListNode o1, ListNode o2) {
                return o1.val - o2.val;
            }
        });

        //遍历每个链表，将所有链表的头结点加入到优先队列中
        for(ListNode listNode : lists){
            if(listNode == null)    //空队列直接跳过
                continue;
            queue.add(listNode);
        }

        //创建dummy node
        ListNode dummyHead = new ListNode(0);
        ListNode cur = dummyHead;

        while(!queue.isEmpty()){
            ListNode nxt = queue.poll(); //取出优先队列的首个节点
            cur.next = nxt;              //将取出的节点挂到合并后的链表中
            cur = cur.next;
            if(nxt.next != null)         //将此节点的下一个节点加入队列
                queue.add(nxt.next);
        }
        return dummyHead.next;
    }