定个小目标之刷LeetCode热题（46）

23. 合并 K 个升序链表

给你一个链表数组，每个链表都已经按升序排列。

请你将所有链表合并到一个升序链表中，返回合并后的链表。

示例 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的增大而增大，该方式的时间复杂度为O(k*k*n)，其推导方式我就不讲了，这里介绍采用优先队列的方式实现，时间复杂度为O(k*n*logk)，思路是，定义一个封装了node的Status对象，其实现Comparable接口，以升序排列所有入队的节点，首先将所有链表的头结点入队，然后一个节点一个节点的构造一个新链表，每出队一个节点后需要入队一个新的节点，代码如下

class Solution {
    class Status implements Comparable<Status> {
        int val;
        ListNode ptr;

        Status(int val, ListNode ptr) {
            this.val = val;
            this.ptr = ptr;
        }

        public int compareTo(Status status2) {
            return this.val - status2.val;
        }
    }

    PriorityQueue<Status> queue = new PriorityQueue<Status>();

    public ListNode mergeKLists(ListNode[] lists) {
        for (ListNode node : lists) {
            if (node != null)
                queue.offer(new Status(node.val, node));
        }
        // 1 4 5; 1 3 4; 2 6
        // 队列: 1 1 2
        // 0-1
        // 队列: 1 2 4
        // 0-1-1
        // 队列: 2 3 4
        // 0-1-1-2
        // 队列: 3 4 6
        // 0-1-1-2-3
        // 队列: 4 4 6
        // 0-1-1-2-3-4
        // 队列: 4 5 6
        // 0-1-1-2-3-4-4
        // 队列: 5 6
        // 0-1-1-2-3-4-4-5
        // 队列: 6
        // 0-1-1-2-3-4-4-5-6
        ListNode head = new ListNode(0);
        ListNode tail = head;
        while (!queue.isEmpty()) {
            Status f = queue.poll();
            tail.next = f.ptr;
            tail = tail.next;
            if (f.ptr.next != null) {
                queue.offer(new Status(f.ptr.next.val, f.ptr.next));
            }
        }
        return head.next;
    }
}

题目链接：题单 - 力扣（LeetCode）全球极客挚爱的技术成长平台