494. 目标和
给你一个非负整数数组 nums 和一个整数 target 。
向数组中的每个整数前添加 ‘+’ 或 ‘-’ ,然后串联起所有整数,可以构造一个 表达式 :
例如,nums = [2, 1] ,可以在 2 之前添加 ‘+’ ,在 1 之前添加 ‘-’ ,然后串联起来得到表达式 “+2-1” 。
返回可以通过上述方法构造的、运算结果等于 target 的不同 表达式 的数目。
示例 1:
输入:nums = [1,1,1,1,1], target = 3
输出:5
解释:一共有 5 种方法让最终目标和为 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
+1 + 1 + 1 + 1 - 1 = 3
示例 2:
输入:nums = [1], target = 1
输出:1
提示:
1 <= nums.length <= 20
0 <= nums[i] <= 1000
0 <= sum(nums[i]) <= 1000
-1000 <= target <= 1000
每个元素有2种可能,所以就深搜,代码如下,但是时间复杂度为 O ( 2 n ) O(2^n) O(2n),比较大。
class Solution {
int dfs(vector<int>& nums, int target, int index) {
if (index >= nums.size()) {
return target == 0 ? 1 : 0;
}
int res = 0;
res += dfs(nums, target - nums[index], index + 1);
res += dfs(nums, target + nums[index], index + 1);
return res;
}
public:
int findTargetSumWays(vector<int>& nums, int target) {
int res = dfs(nums, target, 0);
return res;
}
};
利用背包问题来进行处理,假设我们设取负号的数字的和为 neg
,这样取正数的和就是 sum - neg
,再减去 neg
就有:
( s u m − n e g ) − n e g = t a r g e t (1) (sum-neg)-neg = target\tag{1} (sum−neg)−neg=target(1)
sum和target都是常量,所以就有:
n e g = s u m − t a r g e t 2 (2) neg = \frac{sum-target}{2}\tag{2} neg=2sum−target