题目描述
有一堆石头,每块石头的重量都是正整数。
每一回合,从中选出任意两块石头,然后将它们一起粉碎。假设石头的重量分别为 x 和 y,且 x <= y。那么粉碎的可能结果如下:
如果 x == y,那么两块石头都会被完全粉碎;
如果 x != y,那么重量为 x 的石头将会完全粉碎,而重量为 y 的石头新重量为 y-x。
最后,最多只会剩下一块石头。返回此石头最小的可能重量。如果没有石头剩下,就返回 0。
解题思想
动态规划 ,0-1背包
代码
/*
抽化成一个0-1背包问题:找到尽可能相等的两堆。
sum = sum(stones)
target = sum/2
计算dp[target]:看容量为target的背包尽可能装多少
dp[i]:容量为i的背包装的重量
递推公式:dp[j] = max(dp[j],dp[j-weight[i]]+value[i])
本题:dp[j] = max(dp[j],dp[j-stones[i]]+stones[i])
初始化:dp[j] = 0
遍历顺序:for(int i = 0; i<stones.size();i++)
for(int j = target;j>=stones[i];j--)
*/
class Solution {
public:
int lastStoneWeightII(vector<int>& stones) {
int sum = 0;
for (int a : stones)
sum += a;
int target = sum / 2;
int n = stones.size();
vector<int> dp(target + 1, 0);
for(int i=0; i<n; i++)
for (int j = target; j >= stones[i]; j--) {
dp[j] = max(dp[j], dp[j - stones[i]] + stones[i]);
}
//重量小的一堆为dp[target]
//重量大的一堆为 sum-dp[target]
//让两者相撞 (sum-dp[target])-dp[target]
return (sum - dp[target]) - dp[target];
}
};