链接:力扣
给你一个整数数组 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
分析:
这个和之前的思路相似,我们可以将数组中的数分成两份 left 和 right,如果 left - right = target,那说明就有一种组合方式,我们需要求这种组合方式有几种。我们还可以再简化一下:
right = sum - right
left - (sum - right) = target
left = ( sum + target ) / 2
说明我们要找到数组中能够组合出 ( sum + target ) / 2 这个数的组合方式的个数。
动态规划五部曲:
确定dp数组及下标含义:dp[j] 表示背包容量,能够组合出 j 的组合方式有 dp[j] 种。
递推公式:dp[j] += dp[j - nums[i]] ,当我们新扩张了一个 nums[i] 元素,那么 dp[j] 可以由 dp[j- nums[i]] 到达,也就多了一种组合方式。
初始化:dp[0] = 1, dp[1...n] = 0
遍历顺序:外层循环由小到大,内层循环由后向前
推导:省略
class Solution {
public:
int findTargetSumWays(vector<int>& nums, int target) {
int sum = 0;
for (int num: nums) sum += num;
if (abs(target) > sum) return 0; //这种情况无解
if ((target+sum)%2 == 1) return 0; //这种情况也无解
int bagSize = (target+sum)/2;
vector<int> dp(bagSize+1, 0);
dp[0] = 1;
for (int i = 0; i < nums.size(); ++i) {
for (int j = bagSize; j>=nums[i]; --j) {
dp[j] += dp[j-nums[i]];
}
}
return dp[bagSize];
}
};