有一堆石头,每块石头的重量都是正整数,每一回合,从中选出任意两块石头,然后将它们一起粉碎。假设石头的重量分别为 x 和 y,且 x <= y。那么粉碎的可能结果如下:
- 如果 x == y,那么两块石头都会被完全粉碎;
- 如果 x != y,那么重量为 x 的石头将会完全粉碎,而重量为 y 的石头新重量为 y-x。
最多只会剩下一块石头。返回此石头最小的可能重量。如果没有石头剩下,就返回 0。
思路:将石头分成重量最相近的两小堆,相撞之后石头的重量最小。
转化为01背包问题:物品的重量为stones[i],价值为stones[i]。
dp五部曲:
(1)确定dp数组及下标含义:dp[j]表示容量为j的背包最多可以背的石头
(2)确定递推公式:dp[j] = max(dp[j], dp[j-weight[i]] + value[i]);
(3)dp数组初始化:初始化为0;
(4)确定遍历顺序:内层for循环要倒序;
(5)举例推导dp数组。
实现代码如下:
int lastStoneWeightII(vector<int>& stones) {
int sum = 0;
for(int i = 0; i<stones.size(); i++) {
sum += stones[i];
}
int weight = sum/2;
//1.dp数组下标及含义
//dp[j]:重量为j的背包能背的最大重量
//3.dp数组初始化
vector<int> dp(weight+1, 0);
//4.遍历顺序
//2.确定递推表达式
for(int i = 0; i<stones.size(); i++) {
for(int j = weight; j>=stones[i]; j--) {
dp[j] = max(dp[j], dp[j-stones[i]] + stones[i]);
}
}
return sum - dp[weight] - dp[weight];
}