python优先队列

官方链接: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

你可能感兴趣的:(Python)