000 01背包/中等 LeetCode494. 目标和 NC243 目标和

494. 目标和

描述

给你一个整数数组 nums 和一个整数 target 。
向数组中的每个整数前添加 ‘+’ 或 ‘-’ ，然后串联起所有整数，可以构造一个 表达式 ：
例如，nums = [2, 1] ，可以在 2 之前添加 ‘+’ ，在 1 之前添加 ‘-’ ，然后串联起来得到表达式 “+2-1” 。
返回可以通过上述方法构造的、运算结果等于 target 的不同 表达式 的数目。

分析

假设存在a,b两个数使得a + b = sum,a - b = target。若是在nums中存在和是a的子序列，则可以给nums元素添加正负号来运算等于target。因此题目可转化为求01背包问题：nums中和是a的子序列有多少个。
不同于标准背包问题是求最大质量，本题是求个数，因此设dp[i]为和是i的子序列个数。
dp[i]： 和是i的子序列个数
动态转移方程： dp[i] = dp[i] + dp[i-nums[j]];
动态转移方程有两种情况：1.凑出i的子序列不包括当前数，则继承上一步的结果dp[i]；2.凑出i的子序列包括当前数dp[i-nums[j]]。因为是求个数，所以应该把两个情况加起来。
初始条件： dp[0] = 1;
因为dp[0]是在公式中一切递推结果的起源，如果dp[0]是0的话，递归结果将都是0。
比如dp[3],如nums中有3，dp[3] = dp[3] + dp[3-3]，若dp[0] = 1，dp[3]就能增加1。若dp[0] = 0，每次找到了新的解法就不能增加1了。

class Solution {
    public int findTargetSumWays(int[] nums, int target) {        
        int sum = 0;        
        for(int num : nums){            
            sum += num;        
        }        
        int a = (sum + target) / 2; 
        if(a < 0 || (sum + target) % 2 != 0){
            return 0;
        }       
        int[] dp = new int[a+1]; 
        dp[0] = 1;       
        for(int i = 0; i < nums.length; i++){            
            for(int j = a; j >= nums[i]; j--){                
                dp[j] += dp[j-nums[i]];            
            }        
        }        
        return dp[a];    
    }
}

数组中全部元素加起来为S，题目可理解为用数组中的数凑出-S和S中的一个数。

class Solution {
    public int findTargetSumWays(int[] nums, int target) {        
        int sum = 0;
        for(int num : nums)  {
            sum += num;
        }
        if(target > sum || target < -sum){
            return 0;
        }
        int[][] dp = new int[nums.length + 1][sum*2+1];
        //因为想当target == nums[i]时，dp[target - nums[i]]能为1。
        //由此把0向右平移sum。即dp[0][sum] = 1;
        dp[0][sum] = 1;
        for(int i = 1; i <= nums.length; i++){
            int x = nums[i-1];
            //多个数的和的取值范围是-sum到sum，为了用下标表示和，向右偏移sum。
            for(int j = -sum; j <= sum; j++){
                //加上nums[i-1]的情况
                if((j-x)+sum >= - sum + sum){
                    dp[i][j+sum] += dp[i-1][j-x+sum];
                }
                //减去nums[i-1]的情况
                if((j + x) + sum <= sum + sum){
                    dp[i][j+sum] += dp[i-1][j + x +sum];
                }
            }
        }
        return dp[nums.length][target+sum];
    }
}