Leetcode 1046：最后一块石头的重量（超详细的解法！！！）

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

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

• 如果 x == y，那么两块石头都会被完全粉碎；
• 如果 x != y，那么重量为 x 的石头将会完全粉碎，而重量为 y 的石头新重量为 y-x。

最后，最多只会剩下一块石头。返回此石头的重量。如果没有石头剩下，就返回 0。

提示：

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

解题思路

题目中有非常重要的提示“最重的”，很容易想到通过优先队列来做。将stones中的元素建立一个优先队列，每次从优先队列中弹出两个元素，然后更新优先队列的时候往其中加入y-x即可。由于python中的堆默认是最小堆，所以我在插入元素的时候通过插入对应元素的负值实现。

import heapq
class Solution:
    def lastStoneWeight(self, stones: List[int]) -> int:
        h = []
        for i in stones:
            heapq.heappush(h, -i)
        while len(h) > 1:
            x = heapq.heappop(h)
            y = heapq.heappop(h)
            if x != y:
                heapq.heappush(h, x - y)
        return -h[0] if len(h) == 1 else 0

当然还有一种不错的做法，就是不使用优先队列，而是现将stones排序，这样每次从尾部弹出元素即可，更新元素的时候通过二分法查找y-x要插入的元素位置。

import bisect
class Solution:
    def lastStoneWeight(self, stones: List[int]) -> int:
        stones.sort()
        while len(stones) > 1:
            t1 = stones.pop()
            t2 = stones.pop()
            if t1 != t2:
                bisect.insort_left(stones, t1 - t2)
        return stones[0] if stones else 0

reference:

https://leetcode.com/problems/last-stone-weight/discuss/294956/JavaPython-O(nlogn)

我将该问题的其他语言版本添加到了我的GitHub Leetcode

如有问题，希望大家指出！！！