代码随想录day43 动态规划
题1049 最后一块石头的重量
1,关键的一点是想到将石头分成重量相差最小的两堆,这两堆石头碰撞最后得到的剩余最小。
2,因此抽象成背包问题,设石头的总重量为sum,target= sum/2,那么要找到容量为target的背包能够装下的最大价值。
3,类似于分割等和子集,只是前者需要找到target = dp[target],而本题是需要找到最大价值。
class Solution {
public int lastStoneWeightII(int[] stones) {
int sum = 0;
for(int i: stones) {
sum += i;
}
//寻找重量为target的背包的最大价值
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 - 2 * dp[target];
}
}
题494 目标和
1,假设加法的总和为x,那么减法对应的总和就是sum - x。
所以我们要求的是 x - (sum - x) = target
x = (target + sum) / 2
此时问题就转化为,装满容量为x的背包,有几种方法。
2, 本题是有几种方法装满背包,而不是背包的最大价值或者刚好装满背包,递推公式是累加。
class Solution {
public int findTargetSumWays(int[] nums, int target) {
int sum = 0;
for(int i : nums) {
sum += i;
}
if(Math.abs(target) > sum) return 0;
int plus;//取➕的元素的和
if((target + sum) % 2 != 0 ) return 0;
plus = (target + sum) / 2;
if(plus < 0) plus = -plus;
int[] dp = new int[plus + 1];//有多少种组合方式可以让值累加为plus
dp[0] = 1;
for(int i = 0; i < nums.length; i++) {
for(int j = plus; j >= nums[i]; j--) {
dp[j] = dp[j] + dp[j - nums[i]];
}
}
return dp[plus];
}
}
题474 一和零
1, 如何转换为背包问题:题目要求m个0,n个1的子集的最大元素个数,定义二维dp数组dp[i][j],表示满足i个0 j个1 的子集最多有dp[i][j] 个元素。原数组中的元素相当于物品,依次取出来放入背包。
2,递推公式:对于strs[k],统计出其中0的个数zeroNum, 和1的个数oneNum,若str[i]需要放入背包,则dp[i][j] = dp[i - zeroNum][j - oneNum] + 1, 不放入背包则dp[i][j]保持不变,最后取最值即可。
class Solution {
public int findMaxForm(String[] strs, int m, int n) {
//dp[i][j]数组表示有i个0,j个1时的子集最大的元素个数
int[][] dp = new int[m + 1][n + 1];
int zeroNum, oneNum;
for(String str : strs) {
zeroNum = 0;
oneNum = 0;
for(char c : str.toCharArray()) {
if(c == '0') {
zeroNum++;
} else {
oneNum++;
}
}
for(int i = m; i >= zeroNum; i--) {
for(int j = n; j >= oneNum; j--) {
dp[i][j] = Math.max(dp[i][j], dp[i - zeroNum][j - oneNum] + 1);
}
}
}
return dp[m][n];
}
}