• 周六. 7 月 27th, 2024

5G编程聚合网

5G时代下一个聚合的编程学习网

热门标签

Leetcode1046. 最后一块石头的重量

admin

11 月 28, 2021

1046. 最后一块石头的重量

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

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

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

按照题目意思,每次选出最大的两个数进行操作,采用优先队列进行实现,当队列元素个数大于等于2的时候按照题目要求进行模拟。

最后通过判定队列是否为空决定返回值是队首元素还是零

class Solution {
public:
    int lastStoneWeight(vector<int>& stones) {
        priority_queue<int> q;
        for (int i=0;i<stones.size();i++) q.push(stones[i]);
        while (q.size()>1){
            int tmp1=q.top();
            q.pop();
            int tmp2=q.top();
            q.pop();
            if (tmp1==tmp2) continue;
            else q.push(abs(tmp1-tmp2));
        }
        return q.empty()?0:q.top();
    }
};

发表回复