代码如下:
#encoding=utf-8 print '中国' #插入元素并保持顺序,取得最小的元素 the_list=[2,1,4,3] #堆的特性 data[i] <= data[2*i+1] && data[i] <= data[2*i+2] 二叉树 父节点总是小于子节点 #data[0]因此是最小的 import heapq print the_list heapq.heapify(the_list) print '--heapify' print the_list #取得最小的 print heapq.heappop(the_list) print '--heappop' print the_list #加入新元素 heapq.heappush(the_list,10) heapq.heappush(the_list,3) heapq.heappush(the_list,2) heapq.heappush(the_list,7) heapq.heappush(the_list,8) heapq.heappush(the_list,6) print '--heappush' print the_list #取得代价最高的 利用-1号转化为最小的 class prioq(object): def __init__(self): self.q = [ ] self.i = 0 def push(self, item, cost): heapq.heappush(self.q, (-cost, self.i, item)) self.i += 1 def pop(self): return heapq.heappop(self.q)[-1] the_list=prioq() the_list.push('苹果', 5) the_list.push('西瓜', 3) the_list.push('樱桃', 10) the_list.push('鸭梨', 6) print the_list.pop()
中国
[2, 1, 4, 3]
--heapify
[1, 2, 4, 3]
1
--heappop
[2, 3, 4]
--heappush
[2, 3, 2, 6, 3, 4, 7, 10, 8]
樱桃