给你一个整数数组 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/target-sum
解法:
本题数组长度n最大为20,可以用状态压缩和dfs枚举所有情况暴力解,其时间复杂度都为指数级别O(2^n)。本题可由题意经过公式转化为 从一个正数数组中选取若干元素相加,等于目标值的数目。
设sum为数组元素和,posSum为前面添加正号元素的和,negSum为添加负号元素的和,那么
target = posSum - negSum = sum - negSum - negSum = sum - 2 * negSum;
得 negSum = (sum - target)/ 2;
其中sum与target已知,可求得negSum,要求sum - target为非负偶数,那么就转化为从一个正整数数组中选取若干元素相加,等于目标值的情况有多少种。可以用动态规划做,背包思想。
class Solution {
public:
int findTargetSumWays(vector<int>& nums, int target) {
int sum = 0;
for(auto num : nums)
sum += num;
int t = sum - target;
if(t < 0 || t & 1)
return 0;
int neg = (sum - target) >> 1;
cout<<neg<<endl;
vector<int> dp(neg + 1);
dp[0] = 1;
for(auto num : nums){
for(int j = neg; j >= num; --j){
dp[j] += dp[j - num];
}
}
return dp[neg];
}
};
像这种有明显数学公式的题目,可以通过数学公式进行转化。