最后一块石头的重量

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

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

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

这题我一开始用的是vector排序,然后删除增加指定元素,用了insert,erase.不过最后有问题。报出了地址相关错误

class Solution {
public:
    int lastStoneWeight(vector<int>& stones) {
        sort(stones.begin(),stones.end());
        int tmp;
        while(stones.size()>1)
        {
            int a=stones[stones.size()-1];
            int b=stones[stones.size()-2];
            stones.erase(stones.begin()+(stones.size()-2));
            stones.erase(stones.end());
            if(a!=b)
            {
                int c=a-b;
                for(int i=0;i<stones.size();i++)
                {
                    if(stones[i]>=c)
                    {
                        tmp=i;
                        break;
                    }
                }
                stones.insert(stones.begin()+tmp,c);
            }
        }
        if(stones.size()==0)
        {
            return 0;
        }
        else
        {
            return stones[0];
        }
        
    }
};

第二种可以用优先队列来进行。

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();
    }
};


优先队列默认是大顶堆降序的。

还有一种不用删除元素的办法

class Solution {
public: 
        int lastStoneWeight(vector<int>& stones) {
        int r=stones.size()-1;
        for(;r>=1;r--){
            sort(stones.begin(),stones.end());
            stones[r-1]=stones[r]-stones[r-1];
        }
        return stones[0];
    }
};

但是不是很常规,并且每次都排序或许更浪费时间。不建议

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值