TOPK 问题 描述 如从海量数字中寻找最大的 k 个,这类问题我们称为 TOPK 问题,通常使用堆来解决: 求前 k 大,用最小堆 求前 k 小,用最大堆 例子 现有列表 [1, 2, 0, 3, 5], 求前 2 个大的元素. 如传入列表和 k = 2,输出 [3, 5]. 思路 先放入元素前 k 个建立一个最小堆 迭代剩余元素: 如果当前元素小于堆顶元素,跳过该元素(肯定不是前 k 大) 否则替换堆顶元素为当前元素,并重新调整堆 最后获取 最小堆 中的值,即为 topk 代码如下 impo…