前 K 个高频元素
nums 前 2 高频 [1,2]。
nums 前 2 高频 [1,2]。
堆中保存全部元素,还是只保存当前仍可能进入答案的候选?
频次 map + min-heap k 或 bucket sort。
先说结论:这道题到底解决什么
怎样从“nums 前 2 高频 [1,2]。”推导出 堆 · 前 K 高频,并证明每次状态变化都不会漏掉答案?
中心结论:频次 map + min-heap k 或 bucket sort。
- 1.暴力方案在哪里重复计算,为什么仍然是正确基线?
- 2.堆中保存全部元素,还是只保存当前仍可能进入答案的候选?
- 3.不变量“桶索引=频率可 O(n)。”为什么能保证算法安全前进?
完整题目与题意拆解
给一个非空的数组,输出前 K 个频率最高的元素。
在本站主例中,nums 前 2 高频 [1,2]。
算法最终需要得到或观察:[1,2] 频率最高。
- • 输入:nums 前 2 高频 [1,2]。
- • 机器需要维护:freq map、堆/桶。
- • 最终可观察结果:[1,2] 频率最高。
先看清算法到底要维护什么
先建立输入、目标、输出和第一批状态,不急着进入模板。
频次是堆的排序键
第一层方案:暴力做法
每轮在全部元素中重新寻找最值会重复扫描;堆持续维护下一位最值得处理的候选。
暴力方案的价值是确认题意并提供正确性基线。它通常会覆盖所有候选, 但没有保存已经确认的信息,因此同一状态会被重新计算。
重复工作究竟发生在哪里
把重复读取或重复搜索的区域明确标出,再决定优化必须保存什么。
频次是堆的排序键
整体地图:先做什么,再做什么
- 1建模把输入翻译成“优先级候车区”,明确答案需要观察什么。
- 2状态只维护 freq map、堆/桶。
- 3转移每一步按照 频次 map + min-heap k 或 bucket sort。
- 4收尾读取 [1,2] 频率最高。,并复核边界与复杂度。
优先级候车区:核心概念
先确定堆顶语义,再确定超过容量时淘汰谁。
这道题不是为了记住一个组件名称,而是为了让状态具有可解释的语义:freq map、堆/桶。
- • 桶索引=频率可 O(n)。
建立“优先级候车区”心智模型
用主例建立核心状态,先预测下一步,再公开正确分支和理由。
频次是堆的排序键
核心机制:状态如何一步步变化
这一题是考察优先队列的题目。把数组构造成一个优先队列,输出前 K 个即可。
频次 map + min-heap k 或 bucket sort。
执行过程中持续维护:freq map、堆/桶。
正确性依赖以下不变量:桶索引=频率可 O(n)。
面试时可以压缩为:map+heap 或 bucket O(n)。
落到当前题,执行机制可以压缩为:频次 map + min-heap k 或 bucket sort。每一次更新都必须保持核心不变量,而不是只让样例碰巧得到正确答案。
一次状态转移为什么成立
集中观察一次选择、计算和状态写回,让变量变化与原因同时出现。
当前堆保存频次最高的 1 个
正确性证明:为什么不会漏答案
初始化:算法开始时,全部合法候选仍在状态表示范围内;桶索引=频率可 O(n)。
保持:执行“频次 map + min-heap k 或 bucket sort。”时,只删除已经能证明不可能的候选,并把新信息写回 freq map、堆/桶。
终止:没有待处理状态或达到命中条件时,当前可观察结果就是“[1,2] 频率最高。”。
完整执行过程
- 1题目与输入nums 前 2 高频 [1,2]。 因为:频次 map + min-heap k 或 bucket sort。
- 2先统计频次,再用堆取 TopK哈希计数 O(n)。 因为:TopK 问题先降维到频次。
- 3元素 3 频次 1,维护 Top2 堆按频次维护小根堆。 因为:堆顶是 TopK 中的最低频,可被淘汰。
- 4元素 2 频次 2,维护 Top2 堆按频次维护小根堆。 因为:堆顶是 TopK 中的最低频,可被淘汰。
- 5元素 1 频次 3,维护 Top2 堆按频次维护小根堆。 因为:堆顶是 TopK 中的最低频,可被淘汰。
- 6前 2 高频: 2, 1返回堆中元素。 因为:O(n + m log k),m 为不同元素数。
- 7先统计频次,再用堆取 TopK哈希计数 O(n)。 因为:TopK 问题先降维到频次。
- 8收尾与复杂度[1,2] 频率最高。 因为:时间 O(n) · 空间 O(1)。频次 map + min-heap k 或 bucket sort。
从输入完整走到可观察结果
从主例第一帧运行到答案,时间轴始终显示当前状态、因果解释和下一步。
频次是堆的排序键
把动画和 Go 代码逐行对应
代码窗口不会按容易漂移的固定数字行号硬绑动画,而是根据当前语义阶段, 在完整 Go 实现中定位选择、计算、写回或返回分支。当前变量与高亮行一起变化。
让每个动作都落到 Go 分支
重新执行关键分支,只显示当前代码附近窗口,并说明这行为什么在此刻运行。
频次是堆的排序键
完整 Go 提交代码与最小测试
import "container/heap"
func topKFrequent(nums []int, k int) []int {
m := make(map[int]int)
for _, n := range nums {
m[n]++
}
q := PriorityQueue{}
for key, count := range m {
heap.Push(&q, &Item{key: key, count: count})
}
var result []int
for len(result) < k {
item := heap.Pop(&q).(*Item)
result = append(result, item.key)
}
return result
}
// Item define
type Item struct {
key int
count int
}
// A PriorityQueue implements heap.Interface and holds Items.
type PriorityQueue []*Item
func (pq PriorityQueue) Len() int {
return len(pq)
}
func (pq PriorityQueue) Less(i, j int) bool {
// 注意:因为golang中的heap是按最小堆组织的,所以count越大,Less()越小,越靠近堆顶.
return pq[i].count > pq[j].count
}
func (pq PriorityQueue) Swap(i, j int) {
pq[i], pq[j] = pq[j], pq[i]
}
// Push define
func (pq *PriorityQueue) Push(x interface{}) {
item := x.(*Item)
*pq = append(*pq, item)
}
// Pop define
func (pq *PriorityQueue) Pop() interface{} {
n := len(*pq)
item := (*pq)[n-1]
*pq = (*pq)[:n-1]
return item
}func main() {
// 1. 主例
// 输入:mode="top-k-frequent", nums=[1,1,1,2,2,3], k=2
// 期望:[1,2] 频率最高。
//
// 2. 失败 / 未命中
// 检查:堆法 O(n log k) 通用。
//
// 3. 边界
// 空输入、单元素、最小合法规模,以及答案恰好落在边界的情况。
//
// 4. 迁移
// LC215 第 K 大;LC692 TopK 列表
}Go 参考实现基于 halfrost/LeetCode-Go 的 MIT 许可代码整理,并按本站教学结构补充解释与动画映射。
正确性与复杂度
执行过程中只保留仍可能影响答案的状态。频次 map + min-heap k 或 bucket sort。
额外状态主要用于维护:freq map、堆/桶。
- • 桶索引=频率可 O(n)。
最容易写错的地方
堆法 O(n log k) 通用。
必须额外检查空输入、单元素、未命中或不可达情况,以及恰好落在边界的输入。
最后复盘:带走逻辑链
- 1题意nums 前 2 高频 [1,2]。
- 2重复每轮在全部元素中重新寻找最值会重复扫描;堆持续维护下一位最值得处理的候选。
- 3优化频次 map + min-heap k 或 bucket sort。
- 4证明桶索引=频率可 O(n)。
- 5复杂度时间 O(n),空间 O(1)
- • LC215 第 K 大
- • LC692 TopK 列表