用 python 实现大小堆 & Top K 问题
python的 heapq 模块提供了堆队列(优先队列)的实现,默认是最小堆,即父节点不小于任意一个子节点
使用最小堆可以实现前 K 大的问题,原理是维护一个长度为 K 的堆,每插入一个新元素就弹出之前的最小值,从而最终只剩最大的 K 个值
如想解决前 K 小的问题,则需要将所有元素取负后加入堆,这样最终维持的是所有元素负数中最大的 K 个,再次取负后则得到最小的 K 个元素
案例题目:
1509. Minimum Difference Between Largest and Smallest Value in Three Moves
Top K 问题,找到最大的4个数和最小的4个数,删掉
- 最大的三个
- 最小的三个
- 最大的两个和最小的一个
- 最大的一个和最小的两个
然后用剩下的最大数减最小数,取四种情况中最小的结果返回
def minDifference(self, nums: List[int]) -> int:
if len(nums) <= 4:
return 0
minheap = []
maxheap = []
for num in nums:
if len(minheap