学习动态规划
follow:代码随想录
1049 最后一块石头的重量 II
题目描述:
有一堆石头,用整数数组 stones 表示。其中 stones[i] 表示第 i 块石头的重量。
每一回合,从中选出任意两块石头,然后将它们一起粉碎。假设石头的重量分别为 x 和 y,且 x <= y。那么粉碎的可能结果如下:
如果 x == y,那么两块石头都会被完全粉碎;
如果 x != y,那么重量为 x 的石头将会完全粉碎,而重量为 y 的石头新重量为 y-x。
最后,最多只会剩下一块 石头。返回此石头 最小的可能重量 。如果没有石头剩下,就返回 0。
解答:
class Solution {
public int lastStoneWeightII(int[] stones) {
//该题可以理解成:将石头分成两个重量差不多相等的堆
int sum = 0;
for(int i : stones){
sum += i;
}
//最后看dp[sum / 2]和sum - dp[sum / 2] - dp[sum / 2]
//dp[sum / 2]表示小于等于一半的重量的最大重量,sum - dp[sum/2]就是大于等于一半重量的最小重量
//dp[j]表示重量不超过j的最大重量
int[] dp = new int[sum / 2 + 1];
for(int i = 0 ; i < stones.length; i++){
for(int j = sum / 2; j >= stones[i]; j--){
dp[j] = Math.max(dp[j], dp[j - stones[i]] + stones[i]);
}
}
return sum - dp[sum / 2] - dp[sum / 2];
}
}
494 目标和
题目描述:
给你一个整数数组 nums 和一个整数 target 。
向数组中的每个整数前添加 ‘+’ 或 ‘-’ ,然后串联起所有整数,可以构造一个表达式 :
例如,nums = [2, 1] ,可以在 2 之前添加 ‘+’ ,在 1 之前添加 ‘-’ ,然后串联起来得到表达式 “+2-1” 。
返回可以通过上述方法构造的、运算结果等于 target 的不同表达式的数目。
解析:
回顾之前所练习的三个背包问题:0/1背包问题、分割等和子集、最后一块石头的重量:
0/1背包:给定物品重量,背包重量,物品价值 - 求最大价值
分割等和子集:给定数组中的值(重量),和(target/2),价值也是数值- 求最大价值是否正好为和
最后一块石头的重量:给定石头重量,和(总重量一半)—— 求这个背包能容量的最大容量
针对该题,可以进行简单变形:将该数组分为两部分:”+“为第一部分,”-“为第二部分,具体为:
first - left = target . first + left = sum
可以得到first = (target + sum ) / 2
也就是说,这道题求和为(target + sum ) / 2的集合有多少种。
之前的背包问题都是求最大价值,而这个背包问题求组合方式!
- dp数组的含义
dp[j] = nums,和为j有多少种排列方式 - dp的递推公式
dp[j] = dp[j - nums[i]]
当前要处理的数字为nums[i],想要凑成和为dp[j],所以要看dp[j - nums[i]]有多少种,就证明nums[i]有多少种凑成和为dp[j]的方式,再加上原来的dp[j],就是目前和为j有多少种方式。 - dp的初始化
dp[0] = 1,和为0,只有一种方式:什么也没有 - 遍历顺序
和普通的背包问题一样,先物品再重量,前者正序,后者倒序
class Solution {
public int findTargetSumWays(int[] nums, int target) {
int sum = 0;
for(int i : nums) sum += i;
//证明全正或者全负都无法达到target
if(sum < Math.abs(target)) return 0;
if((target + sum) % 2 != 0) return 0;
int target_new = (target + sum) / 2;
int[] dp = new int[target_new + 1];
dp[0] = 1;
for(int i = 0; i < nums.length; i++){
for(int j = target_new; j >= nums[i]; j--){
dp[j] += dp[j - nums[i]];
}
}
return dp[target_new];
}
}
474. 一和零
题目描述:
给你一个二进制字符串数组 strs 和两个整数 m 和 n 。
请你找出并返回 strs 的最大子集的长度,该子集中 最多 有 m 个 0 和 n 个 1 。
解析:
该题的背包有两个维度m和n,将题目转换为:同时可以放入两个背包的最多物品的数量。
int str_length = 0;
int result = 0;
int count = 0;
public int findMaxForm(String[] strs, int m, int n) {
//动规
//使用二维数组dp[i][j]表示最多包含m个0和n个1的最大子集
int[][] dp = new int[m + 1][n + 1];
//递推公式dp[i][j] = max(dp[i][j], dp[i - m][j - n] + 1)
for(String str : strs){
int zeronum = 0;
int onenum = 0;
for(int i = 0; i < str.length(); i++){
if(str.charAt(i) == '0') zeronum++;
else onenum++;
}
for(int j = m; j >= zeronum; j--){
for(int k = n; k >= onenum; k--){
dp[j][k] = Math.max(dp[j][k], dp[j - zeronum][k - onenum] + 1);
}
}
}
return dp[m][n];
}