官方链接:https://pypi.org/project/sortedcontainers/
最近刷题的时候遇到了优先队列。与一般的队列(先进先出FIFO)不同的是,优先队列一般是最高级的优先出(即最大的先出)。
一开始想到的,可以用堆来维护一个优先队列。不过,python中的堆是一个小根堆,即每次顶部pop出的都是最小值。
这样一来,我想到了41. 数据流中的中位数,能不能把数字变负,写入小根堆。
后来百度的时候发现,其实Python把优先队列封装在了sortedcontainers模块中,(PS:这个模块中还有sorted list, sorted dict, or sorted set,具体说明见顶部官方文档)
优先队列插入数据的方式和list略有不同,采用的是add方法。
举个例子,一目了然
sl = SortedList([2,3,4,5])
sl.add(6)
print(sl)
#SortedList([2, 3, 4, 5, 6])
sl.pop()
print(sl)
#SortedList([2, 3, 4, 5])
优先队列在leetcode中的应用
1675. 数组的最小偏移量
思路:
1.将奇数值*2之后与偶数值一起存入优先队列;
2.由于偶数值不能变大,而奇数不可能进一步变小,因此只需要找到最大奇数值与最小偶数值的差即为结果
from sortedcontainers import SortedList
class Solution(object):
def minimumDeviation(self, nums):
"""
:type nums: List[int]
:rtype: int
"""
#如果是奇数就翻倍
sl = SortedList([num*2 if num&1 else num for num in nums])
res = sl[-1] - sl[0]
#计算差值,只到最大的一个数字是奇数为止
while sl[-1]%2 == 0:
sl.add(sl.pop()//2)
res = min(res,sl[-1]-sl[0])
return res