[LeetCode] Last Stone Weight

1046. Last Stone Weight

You are given an array of integers stones where stones[i] is the weight of the ith stone.

We are playing a game with the stones. On each turn, we choose the heaviest two stones and smash them together. Suppose the heaviest two stones have weights x and y with x <= y. The result of this smash is:

  • If x == y, both stones are destroyed, and
  • If x != y, the stone of weight x is destroyed, and the stone of weight y has new weight y - x.

At the end of the game, there is at most one stone left.

Return the smallest possible weight of the left stone. If there are no stones left, return 0.

1
2
3
4
5
6
7
8
9
10
11
12
13
class Solution {
public:
int lastStoneWeight(vector<int>& s) {
priority_queue<int> pq(begin(s),end(s));
while(pq.size() > 1) {
auto y = pq.top(); pq.pop();
auto x = pq.top(); pq.pop();
if(y > x)
pq.push(y-x);
}
return pq.empty() ? 0 : pq.top();
}
};
Author: Song Hayoung
Link: https://songhayoung.github.io/2022/04/07/PS/LeetCode/last-stone-weight/
Copyright Notice: All articles in this blog are licensed under CC BY-NC-SA 4.0 unless stating additionally.