[100天算法】-目标和（day 79）

题目描述

给定一个非负整数数组，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 位整数存下。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/target-sum
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路 1: DP

简单理解一下题目就是，我们要从数组中选出一个正数集，然后剩下的数字自动变成了一个负数集，这两个集合的和要刚好等于目标数 S。

换句话说，我们要从原数组中选出一个子集，满足元素的和为 target(这个 target 不是原题中的 S)，只要确定这个 target，剩下就是 0-1 背包问题的套路了。

已知：

• 正数集 + 负数集 = S
• 正数集 - 负数集 = sum

sum 是原数组的和。

可得：

• 正数集 = (S + sum) / 2

所以 (S + sum) / 2 就是我们要找的 target。

复杂度

• 时间复杂度：$O(len*(sum+S)/2)$，len 是数组长度，sum 是数组元素和，S 是目标数。
• 空间复杂度：$O((sum+S)/2)$。

代码

JavaScript Code

/**
 * @param {number[]} nums
 * @param {number} S
 * @return {number}
 */
var findTargetSumWays = function (nums, S) {
    const sum = nums.reduce((a, b) => a + b, 0);
    if (sum < S) return 0;

    const sumOfPositives = (sum + S) / 2;
    if (sumOfPositives % 1 !== 0) return 0;

    const dp = Array(sumOfPositives + 1).fill(0);
    // target 为 0 时，正数集为空
    // 也就是只有给所有数字都加上 - 号这一种方法
    dp[0] = 1;
    for (const n of nums) {
        for (let i = sumOfPositives; i >= n; i--) {
            dp[i] = dp[i] + dp[i - n];
        }
    }
    return dp[sumOfPositives];
};

思路 2: DFS

DFS 枚举所有排列组合，计算组合的和，如果满足和等于 S 则结果++。

复杂度

• 时间复杂度：$O(2^n)$，n 是数组长度。
• 空间复杂度：$O(logn)$。

代码

JavaScript Code

/**
 * @param {number[]} nums
 * @param {number} S
 * @return {number}
 */
var findTargetSumWays = function (nums, S) {
    const dfs = (nums, i, sum) => {
        if (sum === S && i === nums.length) return 1;
        if (i > nums.length) return 0;
        return (
            dfs(nums, i + 1, sum + nums[i]) + dfs(nums, i + 1, sum - nums[i])
        );
    };
    return dfs(nums, 0, 0);
};