堆处理 TopK 高频元素

510 字
3 分钟
堆处理 TopK 高频元素

堆处理 TopK 高频元素#

TopK 问题的核心是只保留当前最有价值的 k 个候选。数据量大时,全部排序可能不是最合适的选择。

算法题真正要练的是把题目翻译成状态、边界和转移。模板可以帮助提速,但不能替代对条件的判断。

题型识别#

先统计频次,再维护一个大小为 k 的小根堆。堆顶是当前 TopK 里频次最低的元素,遇到更高频元素就替换。

遇到类似题时,我会先写一句话解释输入、输出和约束,再决定用什么结构。只要这一步说不清,直接写代码大概率会在边界条件上出错。

核心写法#

from collections import Counter
import heapq
def top_k_frequent(nums, k):
heap = []
for num, freq in Counter(nums).items():
heapq.heappush(heap, (freq, num))
if len(heap) > k:
heapq.heappop(heap)
return [num for freq, num in heap]

边界样例#

  • 空输入或只有一个元素时,答案应该是什么。
  • 元素重复时,比较关系是否仍然成立。
  • 结果需要下标、长度、数量,还是具体路径。
  • 时间复杂度能不能从每个元素处理次数解释出来。

复盘#

复盘时要区分求最大 k 个、最小 k 个、频次 TopK 还是数值 TopK。比较键不同,堆里保存的结构也不同。

最小验收样例#

我会至少准备三个样例:一个普通样例,一个边界样例,一个专门卡比较条件的样例。算法代码最怕只过了题目给的样例,却没有覆盖空输入、重复元素或极端顺序。

写完后不要只看答案,还要能解释每一步状态为什么变化。只要能把状态变化讲清楚,换一个题面时就不容易机械套错模板。

小结#

堆处理 TopK 高频元素 的练习重点,是把固定套路拆成可解释的判断。能说明为什么这样维护状态,才算真正掌握。

文章分享

如果这篇文章对你有帮助,欢迎分享给更多人!

堆处理 TopK 高频元素
https://blog.tjcheng.cn/posts/daily/2026-07-08/heap-topk-frequency/
作者
TengJinCheng
发布于
2026-07-08
许可协议
CC BY-NC-SA 4.0
Profile Image of the Author
TengJinCheng
Hello, I'm Teng.
公告
欢迎来到TengJinCheng工作铺!
音乐
封面

音乐

暂未播放

0:000:00
暂无歌词
分类
标签
站点统计
文章
157
分类
7
标签
114
总字数
109,939
运行时长
0 天
最后活动
0 天前
站点信息
构建平台
Local
博客版本
Firefly v6.13.4
文章许可
CC BY-NC-SA 4.0

文章目录