LeetBook刷题笔记022：目标和(Java实现)

LeetBook刷题笔记022：目标和(Java实现)

题目

给你一个整数数组 nums 和一个整数 target 。

向数组中的每个整数前添加 ‘+’ 或 ‘-’ ，然后串联起所有整数，可以构造一个 表达式 ：

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

自己的解：回溯

public int findTargetSumWays(int[] nums, int target) {
    return findAll(nums,0,0,target);
}
public int findAll(int[] nums,int start,int key,int target){
    int n = nums.length;
    if(start==n)
        return key==target?1:0;
    int sum = 0;
    sum += findAll(nums,start+1,key+nums[start],target);
    sum += findAll(nums,start+1,key-nums[start],target);
    return sum;
}

最优解：动态规划

• 将问题转化

  1. 将表达式分为正数和与负数和两部分
  2. 将负数和部分的符号提取出来变成正数和
  3. 联立left - right = target和left + right = sum
  4. 得到left = (sum+target)/2，其中left只能为整数
  5. 转化为求取数组中和为left的一组数的总数的01背包问题

• 套入01背包模板

  价值：nums[i]

  重量：nums[i]

  体积：left

public int findTargetSumWays(int[] nums, int target) {
    int sum = 0;
    for(int x : nums)
        sum += x;
    //所获取的left必须为整数且数组的总数值和必须大于所求数的绝对值
    //不然总数为0
    if((sum+target)%2!=0 || Math.abs(target)>sum)
        return 0;
    sum = (sum+target)/2;
    int n = nums.length;
    int[][] dp = new int[n+1][sum+1];
    //初始化时特别注意有0的情况
    //列为0时可能有多种方法
    //只能初始化行为0的情况(第一列为1,其他列只能为0)
    dp[0][0] = 1;
    //由于初始化了0行数据,故从1开始
    //列只初始化了第一列第一行,故从0开始
    for(int i=1;i<=n;i++){
        for(int t=0;t<=sum;t++){
            //对于dp[i][t]
            dp[i][t] = dp[i-1][t];
            if(t>=nums[i-1])
                dp[i][t] += dp[i-1][t-nums[i-1]];
        }
    }
    return dp[n][sum];
}