1046. 最后一块石头的重量

class Solution:
    def lastStoneWeight(self, stones):
        import heapq
        heap = []
        for i in stones:
            heapq.heappush(heap, i*-1)   #乘以负一，返回一个递增的堆
        left_stones = len(stones)
        while left_stones > 1:
            first = heapq.heappop(heap)#函数heappop弹出最小的元素
            #（总是位于索引0处），并确保剩余元素中最小的那个位于索引0处（保持堆特征）
            second = heapq.heappop(heap)
            if first != second:
                heapq.heappush(heap, first-second)
                left_stones -= 1
            else:
                left_stones -= 2
        if left_stones == 1:
            return heapq.heappop(heap) * (-1)
        else:
            return 0


print(Solution().lastStoneWeight([10, 11, 12]))
print(Solution().lastStoneWeight([2, 2]))

https://leetcode-cn.com/problems/last-stone-weight/

有一堆石头，每块石头的重量都是正整数。

每一回合，从中选出两块最重的石头，然后将它们一起粉碎。假设石头的重量分别为 x 和 y，且 x <= y。那么粉碎的可能结果如下：

如果 x == y，那么两块石头都会被完全粉碎；
如果 x != y，那么重量为 x 的石头将会完全粉碎，而重量为 y 的石头新重量为 y-x。
最后，最多只会剩下一块石头。返回此石头的重量。如果没有石头剩下，就返回 0。

 

提示：

1 <= stones.length <= 30
1 <= stones[i] <= 1000

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/last-stone-weight
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。