【数组-简单】1046. 最后一块石头的重量

最新推荐文章于 2025-12-01 06:10:14 发布

原创最新推荐文章于 2025-12-01 06:10:14 发布 · 274 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #散列表 #c++

leetcode 同时被 2 个专栏收录

714 篇文章

订阅专栏

刷题

643 篇文章

订阅专栏

这篇博客介绍了两种实现方式来解决石头游戏的最后一块石头重量问题。首先展示了Python版本的解决方案，利用排序后选取最大值并进行减法操作。接着，介绍了C++版本的解冑，通过使用最大堆来找出两个最大的元素进行比较和更新。这两个实现都是为了找出当所有石头都两两相消后的剩余重量。

题目
代码
【Python版本-傻瓜式】
在这里插入图片描述

class Solution:
    def lastStoneWeight(self, stones: List[int]) -> int:
        while len(stones)>1:
            stones.sort()
            x=stones[-1]
            stones.pop()
            y=stones[-1]
            stones.pop()
            if x!=y:
                stones.append(abs(x-y))
        if stones:
            return stones[-1]
        return 0

【CPP版本-最大堆】
在这里插入图片描述

class Solution {
public:
    int lastStoneWeight(vector<int>& stones) {
        priority_queue<int> q;
        for (int s: stones) {
            q.push(s);
        }
        while (q.size() > 1) {
            int a = q.top();
            q.pop();
            int b = q.top();
            q.pop();
            if (a > b) {
                q.push(a - b);
            }
        }
        return q.empty() ? 0 : q.top();
    }
};