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