2.队列操作

1.3 保留最后 N 个元素

可以通过collections.deque进行处理，deque(maxlen=N) 构造函数会新建一个固定大小的队列。当新的元素加入并且这个队列已满的时候， 最老的元素会自动被移除掉。

>>> q = deque(maxlen=3)
>>> q.append(1)
>>> q.append(2)
>>> q.append(3)
>>> q
deque([1, 2, 3], maxlen=3)
>>> q.append(4)
>>> q
deque([2, 3, 4], maxlen=3)
>>> q.append(5)

1.4 查找最大或最小的 N 个元素

heapq 模块有两个函数：nlargest() 和nsmallest() 可以找出最大或最小的N个元素

import heapq
nums = [1, 8, 2, 23, 7, -4, 18, 23, 42, 37, 2]
print(heapq.nlargest(3, nums)) # Prints [42, 37, 23]
print(heapq.nsmallest(3, nums)) # Prints [-4, 1, 2]

1.4 使用heapq实现优先队列

import heapq

class Item():
    def __init__(self,name):
        self.name = name
    def __repr__(self):
        return 'Item(\'{}\')'.format(self.name)

class PriorityQueue():
    def __init__(self):
        self.queue = []
        self.index=0 #index 变量的作用是保证同等优先级元素的正确排序
        
    def push(self,item,priority):
        heapq.heappush(self.queue,(priority,self.index,item))
        self.index +=1

    def pop(self):
        value = heapq.heappop(self.queue)
        return value[-1]

if __name__ == '__main__':
    q = PriorityQueue()
    q.push(Item('foo'),1)
    q.push(Item('bar'),5)
    q.push(Item('spam'),4)

    print(q.pop()) # 输出 Item('foo')

heapq模块官方文档
https://docs.python.org/zh-cn/3.6/library/heapq.html