005_007 Python 插入元素并保持顺序,取得最小的元素

代码如下:

#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]
樱桃

你可能感兴趣的:(005_007 Python 插入元素并保持顺序,取得最小的元素)