合并K个升序链表

2021-08-11 11:42:25 477 阅读 AI 写同款 · GEO 优化AI 选题数据监测一键分发 ›

每日一练社区墨衍会员 · AI 创作全网分发关注 + 加入社区

【LeetCode】23. 合并 K 个升序链表最小堆解法是解决多路归并问题的最佳方案，时间复杂度O(N*logk)，空间复杂度O(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

示例 2：

输入：lists = []
输出：[]

示例 3：

输入：lists = [[]]
输出：[]

提示：

k == lists.length
0 <= k <= 10^4
0 <= lists[i].length <= 500
-10^4 <= lists[i][j] <= 10^4
lists[i] 按升序排列
lists[i].length 的总和不超过 10^4

以下程序实现了这一功能，请你填补空白处内容：


public class ListNode {
	int val;
	ListNode next;

	ListNode() {
	}

	ListNode(int val) {
		this.val = val;
	}

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

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

		return merge(lists, 0, lists.length - 1);
	}

	public ListNode merge(ListNode[] lists, int low, int high) {
		if (high - low == 0)
			return lists[low];
		else if (high - low == 1)
			return mergeTwoLists(lists[low], lists[high]);
		else {
			int mid = (low + high) / 2;
			_____________________________;
			return mergeTwoLists(tmp1, tmp2);
		}
	}

	public ListNode mergeTwoLists(ListNode l1, ListNode l2) {
		ListNode head = new ListNode();
		ListNode p = head;
		while (l1 != null && l2 != null) {
			if (l1.val > l2.val) {
				p.next = l2;
				l2 = l2.next;
				p = p.next;
			} else {
				p.next = l1;
				l1 = l1.next;
				p = p.next;
			}
		}
		if (l1 != null)
			p.next = l1;
		if (l2 != null)
			p.next = l2;
		return head.next;
	}
}

...全文