牛刀小试

Posted 2020-12-31 shuai-long

保存最后N个元素

1 from collections import deque
2 #创建一个消息队列,不指定队列大小，会创建一个无限制的队列
3 q = deque(maxlen=N)
4 #方法
5 q.append(num)
6 q.appendleft(num)
7 q.pop(num)
8 q.popleft(num)

从队列两端添加元素或者弹出元素的时间复杂度为O(1)

找到最大或最小的N个元素

import heapq

nums = [1,4,5,2,7,21,45,9,12,-4,16]
#找出nums中最大的N个数
heapq.nlargest(N, nums)
#找出nums中最小的N个数
heapq.nsmallest(N, nums)