Leetcode 1046:最后一块石头的重量(超详细的解法!!!)
in leetcode with 0 comment

Leetcode 1046:最后一块石头的重量(超详细的解法!!!)

in leetcode with 0 comment

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

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

最后,最多只会剩下一块石头。返回此石头的重量。如果没有石头剩下,就返回 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

如有问题,希望大家指出!!!

「如果我的文章对你有很大帮助,那么不妨~!」

coordinate

谢谢老板O(∩_∩)O~

使用微信扫描二维码完成支付

Responses