题目:合并 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;
}
解题思路:
- 创建一个优先队列(最小堆),用于存储每个链表的当前节点。
- 遍历链表数组,将每个链表的头节点加入优先队列。
- 创建一个虚拟头节点
dummy
,用于简化边界条件处理。 - 当优先队列不为空时,执行以下操作:
- 从队列中弹出最小元素,并将其添加到结果链表的末尾。
- 如果弹出元素的下一个节点不为空,将其加入优先队列。
- 返回
dummy.next
,即合并后链表的头节点。
这个问题是一个典型的优先队列应用,它要求我们合并多个已排序的链表。这种类型的问题在处理多个排序数据流合并时非常有用,例如在外部排序和流处理中。