标签:val self 元素 最小 利用 topk minheap iterable
import heapq
class TopK:
'''
思路:
1.先放入元素前k个建立一个最小堆
2.迭代剩余元素:
如果当前元素小于堆顶元素,跳过该元素
否则替换堆顶为当前元素,并重新调整堆
用最小堆来获取最大的几个元素,用前面最小的值替换成最大的值
'''
def __init__(self,iterable,k):
self.minheap = [] #最小堆
self.capacity = k # 最大的几个数
self.iterable = iterable # 传入的列表
def push(self,val): # 传入数据
# 如果超过k个元素就push
if len(self.minheap) >= self.capacity:
min_val = self.minheap[0]
if val < min_val:
pass # 忽略元素
else:
# 替换最小堆元素,并保持最小堆的性质
heapq.heapreplace(self.minheap,val)
else:
# 没有超过k个元素则push
heapq.heappush(self.minheap,val)
def get_topk(self):
for val in self.iterable:
self.push(val) # 出入数据
return self.minheap # 返回最小堆
def test():
import random
i = list(range(1000))
random.shuffle(i)
_ = TopK(i,3)
print(_.get_topk())
test()
标签:val,self,元素,最小,利用,topk,minheap,iterable 来源: https://www.cnblogs.com/wkhzwmr/p/15868605.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。