每日一题
动态规划给我做得头疼,今天依旧是动态规划…
如图
博主也是看了答案后知道了大概思路,但一开始拿到题时知道要用动态规划去做,就是苦于没有思路…还要多加练习啊
拿示例一来分析,每个数都可以去加减,所以是01背包问题的变式 - - - - 问题不在于选不选,而是加和减的问题 ,于是可以得到表格如下
class Solution {
public int findTargetSumWays(int[] nums, int target) {
int sums = 0;
for(int i = 0;i < nums.length;i++){
sums += nums[i];
}
if(Math.abs(target) > Math.abs(sums)){
return 0;
}
int[][] dp = new int[nums.length][sums*2 + 1];
if(nums[0] == 0){
dp[0][sums] = 2;
}else{
dp[0][sums + nums[0]] = 1;
dp[0][sums - nums[0]] = 1;
}
for(int i = 1;i < nums.length;i++){
for(int j = 0;j < sums*2 + 1;j++){
int a = (j + nums[i])<(sums*2+1) ? (j+nums[i]):0;
int b = (j - nums[i]) >= 0 ? (j-nums[i]):0;
dp[i][j] = dp[i-1][a] + dp[i-1][b];
}
}
return dp[nums.length - 1][sums + target];
}
}
这里的DP值就是dp数组的值,我们要定义dp数组肯定要考虑到物品种类与背包容量的问题。dp[ i ][ j ]中i表示有几种物品{数组目前的索引},j就是nums数组元素加减后要得到的目标值。
其中当i = 0时也就是数组中只有一个数时,假设nums[0] = 0 的话,那么dp[0][sum]的值为2(0无论加还是减都是0)。
因为索引不能为负数,所以这里的目标索引值最大取到sum*2+1
状态转移方程:
dp[i][j] = dp[i - 1][j + nums[i]] + dp[i - 1][j - nums[i]]
背包问题要考虑边界:下一个数的加减是否超越了背包的容量或者使背包剩余容量成为负数
官方正解
class Solution {
public int findTargetSumWays(int[] nums, int target) {
int sum = 0;
for (int num : nums) {
sum += num;
}
int diff = sum - target;
if (diff < 0 || diff % 2 != 0) {
return 0;
}
int n = nums.length, neg = diff / 2;
int[][] dp = new int[n + 1][neg + 1];
dp[0][0] = 1;
for (int i = 1; i <= n; i++) {
int num = nums[i - 1];
for (int j = 0; j <= neg; j++) {
dp[i][j] = dp[i - 1][j];
if (j >= num) {
dp[i][j] += dp[i - 1][j - num];
}
}
}
return dp[n][neg];
}
}
作者:LeetCode-Solution
链接:https://leetcode-cn.com/problems/target-sum/solution/mu-biao-he-by-leetcode-solution-o0cp/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
优化后的代码(滚动数组)
class Solution {
public int findTargetSumWays(int[] nums, int target) {
int sum = 0;
for (int num : nums) {
sum += num;
}
int diff = sum - target;
if (diff < 0 || diff % 2 != 0) {
return 0;
}
int neg = diff / 2;
int[] dp = new int[neg + 1];
dp[0] = 1;
for (int num : nums) {
for (int j = neg; j >= num; j--) {
dp[j] += dp[j - num];
}
}
return dp[neg];
}
}
作者:LeetCode-Solution
链接:https://leetcode-cn.com/problems/target-sum/solution/mu-biao-he-by-leetcode-solution-o0cp/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。