解题思路
首先要明确的是,什么情况下最后剩下的石头的重量一定最小?
答案是,把石头尽可能均匀的分成两堆,每次从这里两堆里面拿俩来碰,那么最后剩下的就最小了,那么这道题就转换成了,把一个数据分成两个子数组,要求这两个子数组的数字之和尽量接近,可以用0-1背包来解。
使用动态规划的方法,定义dp[j]含义为:容量为j的背包最大的价值,其中背包容量w和价值v都是石头的重量,那么递归公式为 dp[j]=max(dp[j],dp[j-stones[i]]+stones[i]);
因为使用了一维数组,所以遍历的顺序应该是从后往前,防止前面的数被覆盖
代码
class Solution {
public int lastStoneWeightII(int[] stones) {
int sum=0;
for (var i:stones) sum+=i;
int target=sum/2;
int[] dp=new int[target+1];
for (int i=0;i<stones.length;i++){
for (int j=target;j>=stones[i];j--){
dp[j]=Math.max(dp[j],dp[j-stones[i]]+stones[i]);
}
}
return sum-dp[target]*2;
}
}