|
| 1 | +#### 494. 目标和 |
| 2 | + |
| 3 | +难度:中等 |
| 4 | + |
| 5 | +--- |
| 6 | + |
| 7 | +给你一个非负整数数组 `nums` 和一个整数 `target` 。 |
| 8 | + |
| 9 | +向数组中的每个整数前添加 `'+'` 或 `'-'` ,然后串联起所有整数,可以构造一个 **表达式** : |
| 10 | + |
| 11 | +* 例如,`nums = [2, 1]` ,可以在 `2` 之前添加 `'+'` ,在 `1` 之前添加 `'-'` ,然后串联起来得到表达式 `"+2-1"` 。 |
| 12 | + |
| 13 | +返回可以通过上述方法构造的、运算结果等于 `target` 的不同 **表达式** 的数目。 |
| 14 | + |
| 15 | + **示例 1:** |
| 16 | + |
| 17 | +``` |
| 18 | +输入:nums = [1,1,1,1,1], target = 3 |
| 19 | +输出:5 |
| 20 | +解释:一共有 5 种方法让最终目标和为 3 。 |
| 21 | +-1 + 1 + 1 + 1 + 1 = 3 |
| 22 | ++1 - 1 + 1 + 1 + 1 = 3 |
| 23 | ++1 + 1 - 1 + 1 + 1 = 3 |
| 24 | ++1 + 1 + 1 - 1 + 1 = 3 |
| 25 | ++1 + 1 + 1 + 1 - 1 = 3 |
| 26 | +``` |
| 27 | + |
| 28 | + **示例 2:** |
| 29 | + |
| 30 | +``` |
| 31 | +输入:nums = [1], target = 1 |
| 32 | +输出:1 |
| 33 | +``` |
| 34 | + |
| 35 | + **提示:** |
| 36 | + |
| 37 | +* `1 <= nums.length <= 20` |
| 38 | +* `0 <= nums[i] <= 1000` |
| 39 | +* `0 <= sum(nums[i]) <= 1000` |
| 40 | +* `-1000 <= target <= 1000` |
| 41 | + |
| 42 | +--- |
| 43 | + |
| 44 | +动态规划: |
| 45 | + |
| 46 | +假设所有元素和为 `sum`,那么所有元素都加上负号之后为 `-sum`, 所以元素和的变化值为 `[-sum, sum]`,因此将其范围平移至 `[0, 2 * sum + 1]`。并且定义二维数组 `dp[i][j]` 表示前 `i` 个数组的和为 `j` 的表达式数量。最后的结果为 `dp[len(sum)][sum + target]`。状态转移方程为: |
| 47 | + |
| 48 | +`dp[i][j] = dp[i - 1][j - nums[i - 1]] + dp[i - 1][j + nums[i - 1]]` |
| 49 | + |
| 50 | +```Go |
| 51 | +func findTargetSumWays(nums []int, target int) int { |
| 52 | + n := len(nums) |
| 53 | + sum := 0 |
| 54 | + for _, num := range nums { |
| 55 | + sum += num |
| 56 | + } |
| 57 | + if abs(target) > sum { |
| 58 | + return 0 |
| 59 | + } |
| 60 | + dp := make([][]int, n + 1) |
| 61 | + m := 2 * sum + 1 |
| 62 | + for i := range dp { |
| 63 | + dp[i] = make([]int, m + 1) |
| 64 | + } |
| 65 | + dp[0][sum] = 1 |
| 66 | + for i := 1; i <= n; i++ { |
| 67 | + for j := 0; j <= m; j++ { |
| 68 | + dp[i][j] = 0 |
| 69 | + if j - nums[i - 1] >= 0{ |
| 70 | + dp[i][j] += dp[i - 1][j - nums[i - 1]] |
| 71 | + } |
| 72 | + if j + nums[i - 1] <= m { |
| 73 | + dp[i][j] += dp[i - 1][j + nums[i - 1]] |
| 74 | + } |
| 75 | + } |
| 76 | + } |
| 77 | + return dp[n][target + sum] |
| 78 | +} |
| 79 | + |
| 80 | +func abs(a int) int { |
| 81 | + if a < 0 { |
| 82 | + return -a |
| 83 | + } |
| 84 | + return a |
| 85 | +} |
| 86 | +``` |
0 commit comments