最后一块石头的重量
学习记录自代码随想录
要点:1.想到求最后剩下石头的重量可以转化为求石头组合重量最接近石头重量总和的一半和此时剩下石头重量的差值,从而联想到Leetcode 416此题便可以相同套路求解
class Solution {
public:
int lastStoneWeightII(vector<int>& stones) {
int sum = 0;
int target = 0;
for(int a : stones) sum += a;
target = sum / 2;
// 1.dp[j]为背包容量为j时最大价值dp[j]
vector<int> dp(1501, 0);
// 2.递推公式dp[j] = max(dp[j], dp[j-stones[i]]+stones[i])
// 3.dp数组初始化为0
// 4.遍历顺序,先遍历物品再反向遍历容量
for(int i = 0; i < stones.size(); i++){
for(int j = target; j >= stones[i]; j--){
dp[j] = max(dp[j], dp[j-stones[i]] + stones[i]);
}
}
return abs(dp[target] - (sum-dp[target]));
}
};