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

菜菜2022

于 2022-02-21 19:28:34 发布

阅读量265

点赞数

CC 4.0 BY-SA版权

分类专栏：刷题 # leetcode 文章标签： leetcode 散列表 c++

本文链接：https://blog.youkuaiyun.com/kz_java/article/details/123053718

leetcode 同时被 2 个专栏收录

714 篇文章

订阅专栏

刷题

643 篇文章

订阅专栏

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

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目
代码
【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();
    }
};