[LeetCode 中等 动态规划]494. 目标和

题目描述

给定一个非负整数数组，a1, a2, …, an, 和一个目标数，S。现在你有两个符号 + 和 -。对于数组中的任意一个整数，你都可以从 + 或 -中选择一个符号添加在前面。

返回可以使最终数组和为目标数 S 的所有添加符号的方法数。

示例：

输入：nums: [1, 1, 1, 1, 1], S: 3
输出：5
解释：

-1+1+1+1+1 = 3
+1-1+1+1+1 = 3
+1+1-1+1+1 = 3
+1+1+1-1+1 = 3
+1+1+1+1-1 = 3

一共有5种方法让最终目标和为3。

提示：

数组非空，且长度不会超过 20 。
初始的数组的和不会超过 1000 。
保证返回的最终结果能被 32 位整数存下。

蛮力+递归

class Solution {
    int count =0;
    public int findTargetSumWays(int[] nums, int S) {
        getRes(nums,0,0,S);
        return count;
    }

    public void getRes(int[] nums,int index, int res, int S){
        if(index==nums.length && res == S)  count++;

        if(index < nums.length){
            getRes(nums,index+1,res+nums[index],S);
            getRes(nums,index+1,res-nums[index],S);
        }
    }
}

动态规划

经典背包问题
class Solution {
    public int findTargetSumWays(int[] nums, int S) {
            if(S>1000 || S<-1000) return 0;
            int len = nums.length;
            int[][] dp = new int[len][2001];
            dp[0][1000 + nums[0]] = 1;
            //非常重要 这里需要改写成 += 因为可能出现 第一个元素为0的状况
            dp[0][1000 - nums[0]] += 1;
            
            for(int i=1;i<len;i++){
                for (int sum = -1000; sum <= 1000; sum++) {
                    if(dp[i-1][1000+sum] != 0){
                        dp[i][1000+sum - nums[i]] += dp[i-1][1000+sum];
                        dp[i][1000+sum + nums[i]] += dp[i-1][1000+sum];
                    }
               }
            }
            return dp[len-1][1000+S];
    }

}

动态规划+空间优化

class Solution {
    public int findTargetSumWays(int[] nums, int S) {
            if(S>1000 || S<-1000) return 0;
            int len = nums.length;
            int[] dp = new int[2001];
            dp[1000 + nums[0]] = 1;
            //非常重要 这里需要改写成 += 因为可能出现 第一个元素为0的状况
            dp[1000 - nums[0]] += 1;
            
            for(int i=1;i<len;i++){
                int[] temp = new int[2001];
                for (int sum = -1000; sum <= 1000; sum++) {
                    if(dp[1000+sum] != 0){
                        if(1000+sum - nums[i]>=0) temp[1000+sum - nums[i]] += dp[1000+sum];
                        if(1000+sum + nums[i]<2001) temp[1000+sum + nums[i]] += dp[1000+sum];
                    }
               }
               // 这里要复制是因为 有排序要求 同一个数字加或者减 两种情况 只用一个数组会影响结果
               dp = temp;
            }
            return dp[1000+S];
    }

}