2019.12.3 LEETCODE AG

最新推荐文章于 2020-02-16 13:13:53 发布

Emily_20191106

最新推荐文章于 2020-02-16 13:13:53 发布

阅读量90

点赞数

本文链接：https://blog.csdn.net/weixin_44227332/article/details/103379841

版权

堆——简单
贪心算法——简单

1046 最后一块石头的重量

有个库叫 import heapq

有几个用法

import heapq #生成最小堆优先队列
heapq.heapify(object)
heapq.heappop(object) #pop 最小值
heapq.heappush(object，var) #将元素push队列


class Solution:
    def lastStoneWeight(self, stones) :
        import heapq
        stone_heapq = [-i for i in stones]
        heapq.heapify(stone_heapq)
        while len(stone_heapq)>1:
            a = heapq.heappop(stone_heapq)
            b = heapq.heappop(stone_heapq)
            if a<b:
                heapq.heappush(stone_heapq,a-b)
        if stone_heapq:
            res = -stone_heapq[0]
        else:
            res = 0
        return res

703 数据流中第K大元素

关于 heap 包的用法

https://docs.python.org/zh-cn/3/library/heapq.html

class KthLargest:

    def __init__(self, k: int, nums: List[int]):
        self.nums = heapq.nlargest(k, nums)[::-1]  ## 把nums的最大的k个数从小到大排序。注意如果nums=k-1的话，heapq.nlargest也是有效的，会把k-1个数从小到大排序。
        self.k = k  ## 用self.k记录k

    def add(self, val: int) -> int:
        if len(self.nums) == self.k:  ## 如果是k个数
            if val > self.nums[0]:  ## 如果val大于k个数中最小的
                    heapq.heapreplace(self.nums, val)  ##那么丢掉nums[0],把val塞进堆里。
            return self.nums[0]  ## 返回最小的数
        heapq.heappush(self.nums, val)  ## 如果是k-1个数，把val塞进堆里
        return self.nums[0]  ## 返回最小的数

作者：erik_chen
链接：https://leetcode-cn.com/problems/kth-largest-element-in-a-stream/solution/yong-zui-xiao-dui-shi-xian-by-erik_chen/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。