给你一个整数数组 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
思考
回溯法
我们可以求得总数sum,所以我们只要减去一些元素后sum==target即求得一种方法。
sum = sum - (nums[i]*2);
为什么每次对sum减去一个数*2?
因为本来sum中包含这个数
if(nums[i]==0) continue;
这句话是跳过0的情况,我们可以提前算出0的个数n,最后在跳过0的基础上求得方法个数乘以2的n次方即可。
因为 0前面加±对结果没影响。
class Solution {
int sum = 0;
int method = 0;
public int findTargetSumWays(int[] nums, int target) {
int zeroCount = 0;
for (int i=0;i<nums.length;i++){
if(nums[i]==0)
zeroCount++;
sum = sum + nums[i];
}
int startIndex = 0;
backTracking(startIndex,nums,target);
return (int) (method*Math.pow(2,zeroCount));
}
private void backTracking(int startIndex, int[] nums, int target) {
if(sum==target){
method++;
return;
}
for (int i=startIndex;i<nums.length;i++){
if(nums[i]==0) continue;
if(sum-(nums[i]*2)<target){
continue;
}
sum = sum - (nums[i]*2);
backTracking(i+1,nums,target);
sum = sum + (nums[i]*2);
}
}
}