python使用heapq实现小顶堆(TopK大)/大顶堆(BtmK小)
来源:互联网 发布:黄瓜什么意思网络用语 编辑:程序博客网 时间:2024/05/20 01:10
参考链接
https://www.coder4.com/archives/3844
求一个数列前K大数的问题经常会遇到,在程序中一般用小顶堆可以解决,下面的代码是使用python的heapq实现的小顶堆示例代码:
# !/usr/bin/env python # -*- coding:gbk -*- import sys import heapq class TopKHeap(object): def __init__(self, k): self.k = k self.data = [] def push(self, elem): if len(self.data) < self.k: heapq.heappush(self.data, elem) else: topk_small = self.data[0] if elem > topk_small: heapq.heapreplace(self.data, elem) def topk(self): return [x for x in reversed([heapq.heappop(self.data) for x in xrange(len(self.data))])] def main(): list_num = [1, 2, 3, 4, 5, 6, 7, 8, 9] th = TopKHeap(5) for i in list_num: th.push(i) print th.topk() if __name__ == "__main__": main()
python的heapq在实现的时候,没有像STL或者Java可以传入比较函数,具体的原因可以参考参考文档给出的链接。
因此有些人想出了比较trick的思路。一句话概括如下:
push(e)改为push(-e),pop(e)为-pop(e),也就是说存入和取出的数都是相反数,其他逻辑和TopK相同。(点赞)
实现用户自定义的比较函数,允许elem是一个tuple,按照tuple的第一个元素进行比较,所以可以把tuple的第一个元素作为我们的比较的key。
英文原话:
The heapq documentation suggests that heap elements could be tuples in which the first element is the priority and defines the sort order.
import heapq class MyHeap(object): def __init__(self, initial=None, key=lambda x:x): self.key = key if initial: self._data = [(key(item), item) for item in initial] heapq.heapify(self._data) else: self._data = [] def push(self, item): heapq.heappush(self._data, (self.key(item), item)) def pop(self): return heapq.heappop(self._data)[1]
0 0
- python使用heapq实现小顶堆(TopK大)/大顶堆(BtmK小)
- Python的heapq模块实现大顶堆,小顶堆
- TopK---返回第K小(大)的数字
- Python 小技巧——<Cookbook>读书笔记(2)deque和heapq
- Java解决TopK问题(使用集合和直接实现)
- 一题多解(五) —— topK(数组中第 k 大/小的数)
- python学习笔记(十三)标准库heapq
- python heapq
- python heapq
- python-heapq
- Python利用heapq实现一个优先级队列
- python模块heapq之简单学习使用
- Spark下的TopK实现(Java)
- 矩阵topK 给一个矩阵,各行各列均有序,求第k小值。(k大的值)
- Python 数据结构与算法——选取算法(TopK)
- 利用python的heapq实现prim算法以及优先级队列
- [Python] heapq简介(转载自:http://www.cnblogs.com/lexus/p/3324991.html)
- Python -- 堆数据结构 heapq
- hduCoconuts(离散化)
- android 模拟触摸板控制鼠标(解决小屏幕控制大屏幕)
- Mybatis高级应用-嵌套查询association和collection
- 屏幕适配的前世今生
- 第三十六套
- python使用heapq实现小顶堆(TopK大)/大顶堆(BtmK小)
- android:layout_height="match_parent"和android:layout_marginTop="100dp"同时作用
- DevExpress 编辑GridControl中合并单元格
- Bootstrap 网格系统
- Tinker热修复学习笔记
- 团战可以输、提莫必须死
- bgsave出错原因分析
- JavaScript基础
- 【OA】——hibernate映射关系编写技巧