合并 K 个排序链表——Java解答

题目：合并 K 个排序链表

题目描述：

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

示例：

假设有以下三个链表：

1->4->5,
1->3->4,
2->6

合并后的链表为 1->1->2->3->4->4->5->6。

Java 解答：

public ListNode mergeKLists(ListNode[] lists) {
    if (lists == null || lists.length == 0) return null;

    PriorityQueue<ListNode> queue = new PriorityQueue<>(lists.length, (a, b) -> a.val - b.val);

    ListNode dummy = new ListNode(0);
    ListNode tail = dummy;

    for (ListNode node : lists) {
        if (node != null) {
            queue.add(node);
        }
    }

    while (!queue.isEmpty()) {
        tail.next = queue.poll();
        tail = tail.next;

        if (tail.next != null) {
            queue.add(tail.next);
        }
    }

    return dummy.next;
}

解题思路：

1. 创建一个优先队列（最小堆），用于存储每个链表的当前节点。
2. 遍历链表数组，将每个链表的头节点加入优先队列。
3. 创建一个虚拟头节点 dummy，用于简化边界条件处理。
4. 当优先队列不为空时，执行以下操作：
   • 从队列中弹出最小元素，并将其添加到结果链表的末尾。
   • 如果弹出元素的下一个节点不为空，将其加入优先队列。
5. 返回 dummy.next，即合并后链表的头节点。

这个问题是一个典型的优先队列应用，它要求我们合并多个已排序的链表。这种类型的问题在处理多个排序数据流合并时非常有用，例如在外部排序和流处理中。