LC23 合并 K 个升序链表
K=3 条升序链表 [[1,4,5],[1,3,4],[2,6]],合并为一条升序链表;先堆理解,再分治面试写法。
K=3 条升序链表 [[1,4,5],[1,3,4],[2,6]],合并为一条升序链表;先堆理解,再分治面试写法。
堆中保存全部元素,还是只保存当前仍可能进入答案的候选?
K 路归并:堆维护各路当前头;Go 面试推荐分治 bottom-up merge,step=1/2/4 两两合并到 lists[0]。
先说结论:这道题到底解决什么
怎样从“K=3 条升序链表 [[1,4,5],[1,3,4],[2,6]],合并为一条升序链表;先堆理解,再分治面试写法。”推导出 链表 / K 路归并 / 小根堆 / 分治 / Go,并证明每次状态变化都不会漏掉答案?
中心结论:K 路归并:堆维护各路当前头;Go 面试推荐分治 bottom-up merge,step=1/2/4 两两合并到 lists[0]。
- 1.暴力方案在哪里重复计算,为什么仍然是正确基线?
- 2.堆中保存全部元素,还是只保存当前仍可能进入答案的候选?
- 3.不变量“结果链表始终升序。”为什么能保证算法安全前进?
完整题目与题意拆解
K=3 条升序链表 [[1,4,5],[1,3,4],[2,6]],合并为 1→1→2→3→4→4→5→6;先堆理解,再分治面试写法。
在本站主例中,K=3 条升序链表 [[1,4,5],[1,3,4],[2,6]],合并为 1→1→2→3→4→4→5→6;先堆理解,再分治面试写法。
算法最终需要得到或观察:1→1→2→3→4→4→5→6
- • 输入:K=3 条升序链表 [[1,4,5],[1,3,4],[2,6]],合并为一条升序链表;先堆理解,再分治面试写法。
- • 机器需要维护:堆(候选头)、tail(总装线尾)、各路 head 指针。
- • 最终可观察结果:1→1→2→3→4→4→5→6
先看清算法到底要维护什么
先建立输入、目标、输出和第一批状态,不急着进入模板。
第一层方案:暴力做法
每轮在全部元素中重新寻找最值会重复扫描;堆持续维护下一位最值得处理的候选。
暴力方案的价值是确认题意并提供正确性基线。它通常会覆盖所有候选, 但没有保存已经确认的信息,因此同一状态会被重新计算。
重复工作究竟发生在哪里
把重复读取或重复搜索的区域明确标出,再决定优化必须保存什么。
整体地图:先做什么,再做什么
- 1建模把输入翻译成“优先级候车区”,明确答案需要观察什么。
- 2状态只维护 堆(候选头)、tail(总装线尾)、各路 head 指针。
- 3转移每一步按照 K 路归并:堆维护各路当前头;Go 面试推荐分治 bottom-up merge,step=1/2/4 两两合并到 lists[0]。
- 4收尾读取 1→1→2→3→4→4→5→6,并复核边界与复杂度。
优先级候车区:核心概念
先确定堆顶语义,再确定超过容量时淘汰谁。
这道题不是为了记住一个组件名称,而是为了让状态具有可解释的语义:堆(候选头)、tail(总装线尾)、各路 head 指针。
- • 结果链表始终升序。
- • 堆中元素 = 各条轨道当前暴露的头(最多 K 个)。
- • 堆顶 = 下一轮要放行的全局最小候选。
建立“优先级候车区”心智模型
用主例建立核心状态,先预测下一步,再公开正确分支和理由。
核心机制:状态如何一步步变化
借助分治的思想,把 K 个有序链表两两合并即可。相当于是第 21 题的加强版。
K 路归并:堆维护各路当前头;Go 面试推荐分治 bottom-up merge,step=1/2/4 两两合并到 lists[0]。
执行过程中持续维护:调度塔堆、tail 总装线尾、各路 head 指针。
正确性依赖以下不变量:结果链表始终升序。;堆中只保存各轨当前暴露的头(最多 K 个)。;堆顶是下一轮要放行的全局最小候选。
面试时可以压缩为:小根堆维护 K 路头,O(N log K)/O(K);也可分治 merge。pop 后只 push 该路 next。
落到当前题,执行机制可以压缩为:K 路归并:堆维护各路当前头;Go 面试推荐分治 bottom-up merge,step=1/2/4 两两合并到 lists[0]。每一次更新都必须保持核心不变量,而不是只让样例碰巧得到正确答案。
一次状态转移为什么成立
集中观察一次选择、计算和状态写回,让变量变化与原因同时出现。
正确性证明:为什么不会漏答案
初始化:算法开始时,全部合法候选仍在状态表示范围内;结果链表始终升序。
保持:执行“K 路归并:堆维护各路当前头;Go 面试推荐分治 bottom-up merge,step=1/2/4 两两合并到 lists[0]。”时,只删除已经能证明不可能的候选,并把新信息写回 堆(候选头)、tail(总装线尾)、各路 head 指针。
终止:没有待处理状态或达到命中条件时,当前可观察结果就是“1→1→2→3→4→4→5→6”。
完整执行过程
- 1输入 [[1,4,5],[1,3,4],[2,6]]每条链表只是一串按编号递增的矿车;还不能直接比较「整轨」谁更小。 因为:升序链表保证:每条轨道内部从左到右编号单调不减。
- 2每轮 O(K) 扫描调度员用肉眼扫三条轨道车头,找编号最小的矿车。 因为:做法正确但慢:重复扫描浪费,需要能 O(log K) 取最小的结构。
- 3pop 1(L0)调度塔放行最小编号矿车 1,挂到总装线末尾(复用原节点指针)。 因为:堆顶是全局最小候选;接到 tail 后结果仍升序。
- 4pop 1(L1)调度塔放行最小编号矿车 1,挂到总装线末尾(复用原节点指针)。 因为:堆顶是全局最小候选;接到 tail 后结果仍升序。
- 5push 6(L2)只有被弹出的那条轨道 L2 需要补入下一辆;其它路已在堆中。 因为:堆始终 = 各路当前头的集合,大小仍 ≤ K。
- 6pop 4(L0)堆顶 4(L0) 弹出。L0 还剩 5 待入堆。 因为:堆顶始终是剩余候选中的最小编号。
- 7pop 6(L2),L2 耗尽堆顶 6(L2) 弹出后 L2 耗尽。堆即将变空。 因为:堆顶始终是剩余候选中的最小编号。
- 8O(N log K) · O(K)复述三步:只看 K 个头 → 堆维护候选 → pop 后只 push 该路 next。 因为:每个节点入堆出堆各一次;堆高 O(log K)。
从输入完整走到可观察结果
从主例第一帧运行到答案,时间轴始终显示当前状态、因果解释和下一步。
把动画和 Go 代码逐行对应
代码窗口不会按容易漂移的固定数字行号硬绑动画,而是根据当前语义阶段, 在完整 Go 实现中定位选择、计算、写回或返回分支。当前变量与高亮行一起变化。
让每个动作都落到 Go 分支
重新执行关键分支,只显示当前代码附近窗口,并说明这行为什么在此刻运行。
完整 Go 提交代码与最小测试
// ListNode define
/**
* Definition for singly-linked list.
* type ListNode struct {
* Val int
* Next *ListNode
* }
*/
func mergeKLists(lists []*ListNode) *ListNode {
length := len(lists)
if length < 1 {
return nil
}
if length == 1 {
return lists[0]
}
num := length / 2
left := mergeKLists(lists[:num])
right := mergeKLists(lists[num:])
return mergeTwoLists1(left, right)
}
func mergeTwoLists1(l1 *ListNode, l2 *ListNode) *ListNode {
if l1 == nil {
return l2
}
if l2 == nil {
return l1
}
if l1.Val < l2.Val {
l1.Next = mergeTwoLists1(l1.Next, l2)
return l1
}
l2.Next = mergeTwoLists1(l1, l2.Next)
return l2
}func main() {
// 1. 主例
// 输入:mode="merge-k-lists", lists=[[1,4,5],[1,3,4],[2,6]]
// 期望:1→1→2→3→4→4→5→6
//
// 2. 失败 / 未命中
// 检查:把所有节点塞进堆。
//
// 3. 边界
// 空输入、单元素、最小合法规模,以及答案恰好落在边界的情况。
//
// 4. 迁移
// LC21 合并两个有序链表;LC347 Top K
}Go 参考实现基于 halfrost/LeetCode-Go 的 MIT 许可代码整理,并按本站教学结构补充解释与动画映射。
正确性与复杂度
执行过程中只保留仍可能影响答案的状态。K 路归并:堆维护各路当前头;Go 面试推荐分治 bottom-up merge,step=1/2/4 两两合并到 lists[0]。
额外状态主要用于维护:堆(候选头)、tail(总装线尾)、各路 head 指针。
- • 结果链表始终升序。
- • 堆中元素 = 各条轨道当前暴露的头(最多 K 个)。
- • 堆顶 = 下一轮要放行的全局最小候选。
最容易写错的地方
误区:把所有节点都塞进堆 —— 只需每条链表当前暴露的头节点。
误区:堆里只存 Val 不存 *ListNode —— 会丢失 next,无法接链。
误区:忘记处理空链表 —— 初始化时 head==nil 不要 Push。
误区:弹出后忘记 Push node.Next —— 该轨道下一辆矿车进不了候选区。
误区:相等值比较不稳定 —— 用链表 id 作 tie-breaker。
必须额外检查空输入、单元素、未命中或不可达情况,以及恰好落在边界的输入。
最后复盘:带走逻辑链
- 1题意K=3 条升序链表 [[1,4,5],[1,3,4],[2,6]],合并为一条升序链表;先堆理解,再分治面试写法。
- 2重复每轮在全部元素中重新寻找最值会重复扫描;堆持续维护下一位最值得处理的候选。
- 3优化K 路归并:堆维护各路当前头;Go 面试推荐分治 bottom-up merge,step=1/2/4 两两合并到 lists[0]。
- 4证明结果链表始终升序。;堆中元素 = 各条轨道当前暴露的头(最多 K 个)。;堆顶 = 下一轮要放行的全局最小候选。
- 5复杂度时间 O(N log K),空间 O(K)
- • LC21 合并两个有序链表
- • LC347 Top K