Back to all solutions
#1049 - Last Stone Weight II
Problem Description
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.
Solution
/**
* @param {number[]} stones
* @return {number}
*/
var lastStoneWeightII = function(stones) {
const totalSum = stones.reduce((sum, weight) => sum + weight, 0);
const target = Math.floor(totalSum / 2);
const set = new Set([0]);
for (const stone of stones) {
const prev = new Set(set);
for (const sum of prev) {
if (sum + stone <= target) {
set.add(sum + stone);
}
}
}
return totalSum - 2 * Math.max(...set);
};