代码如下:
#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]
樱桃