[LeetCode] Last Stone Weight II

1049. Last Stone Weight II

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 any two stones and smash them together. Suppose the 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
14
15
16
17
class Solution {
int dp[1500];
public:
int lastStoneWeightII(vector<int>& S) {
bitset<1501> dp{1};
int sum = 0;
for(auto& s : S) {
sum += s;
for(int i = min(1500, sum); i >= s; i--)
dp[i] = dp[i] + dp[i-s];
}
for(int i = sum / 2; i >= 0; i--) {
if(dp[i]) return sum - i*2;
}
return -1;
}
};
Author: Song Hayoung
Link: https://songhayoung.github.io/2022/04/07/PS/LeetCode/last-stone-weight-ii/
Copyright Notice: All articles in this blog are licensed under CC BY-NC-SA 4.0 unless stating additionally.