leetcode 494. 目标和

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 位整数存下

---

使用回溯算法，[1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1]，0这个例子会超时 =-=

class Solution:
    def findTargetSumWays(self, nums: List[int], S: int) -> int:
        self.cnt = 0
        if not nums: return 0
        n = len(nums)
        def backtrack(count, index, res):
            if count == n and res == S:
                self.cnt += 1
                return 
            for i in range(index, n):
                backtrack(count+1, i+1, res+nums[i])
                backtrack(count+1, i+1, res-nums[i])
        backtrack(0, 0, 0)
        return self.cnt

---

修改了一下，但是还是超出限制，坑=-=，只有换方法了

class Solution:
    def findTargetSumWays(self, nums: List[int], S: int) -> int:
        self.cnt = 0
        if not nums: return 0
        n = len(nums)
        def backtrack(i, res):
            if i == n:
                if res == S:
                    self.cnt += 1
                return 
            else:
                backtrack(i+1, res+nums[i])
                backtrack(i+1, res-nums[i])
        backtrack(0, 0)
        return self.cnt

---

下面是一种数学的方法：

• 开辟一个长度为P+1的数组，命名为dp
• dp的第x项，代表组合成数字x有多少方法。比如说,dp[0] = 1，代表组合成0只有1中方法，即什么也不取。比如说dp[5] = 3 ，代表使总和加到5总共有三种方法。
• 所以最后返回的就是dp[P]，代表组合成P的方法有多少种

---

• 首先排除几种不成立的情况。
• 这里的循环有两种方式：第一个for 是列表内的数字num，第二是从target 到 num-1取值。
• dp[ i ] = dp[ i ] + dp[ i -num]，含义数字 i 的组合方式等于之前的取的方式，加上本次不取num的组合数相加。
• 每次取一个数字num，对dp[j]的j∈[num, target]内，如果dp[j-num]有值，那么dp[j-num]+num就是dp[j],就可以算是一种组合。所以每次首先遍历nums中的num,看是否存在差值为num，可以从dp[j-num]更新得到dp[j].

/* 计算 nums 中有几个子集的和为 sum */
int subsets(int[] nums, int sum) {
    int n = nums.length;
    int[][] dp = new int[n + 1][sum + 1];
    // base case
    for (int i = 0; i <= n; i++) {
        dp[i][0] = 1;
    }
    
    for (int i = 1; i <= n; i++) {
        for (int j = 0; j <= sum; j++) {
            if (j >= nums[i-1]) {
                // 两种选择的结果之和
                dp[i][j] = dp[i-1][j] + dp[i-1][j-nums[i-1]];
            } else {
                // 背包的空间不足，只能选择不装物品 i
                dp[i][j] = dp[i-1][j];
            }
        }
    }
    return dp[n][sum];
}

作者：labuladong
链接：https://leetcode-cn.com/problems/target-sum/solution/dong-tai-gui-hua-he-hui-su-suan-fa-dao-di-shui-shi/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

class Solution:
    def findTargetSumWays(self, nums: List[int], S: int) -> int:
        target = S + (t:=sum(nums))
        if t < S or target < 0 or target % 2: return 0
        target //= 2
        dp = [0] * (target + 1)
        dp[0] = 1
        for num in nums:
            for i in range(target, num - 1, -1):
                if i - num >=  0: 
                    dp[i] += dp[i - num]
        return dp[-1]

作者：powcai
链接：https://leetcode-cn.com/problems/target-sum/solution/dong-tai-gui-hua-by-powcai-22/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。