题目
给你一个整数数组 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
方法
深度优先遍历
遍历所有情况,最后判断target是否抵消为0,若为0则记录一次
- 时间复杂度:O(2n),n为nums数组的长度,每位有两种情况
- 空间复杂度:O(n),空间的占用取决于递归栈的长度,即数组的长度
class Solution {
private int count = 0;
public int findTargetSumWays(int[] nums, int target) {
calculate(nums,target,0);
return count;
}
private void calculate(int[] nums,int target,int depth){
if(depth==nums.length){
if(target==0){
count++;
}
return;
}
calculate(nums,target-nums[depth],depth+1);
calculate(nums,target+nums[depth],depth+1);
}
}
动态规划?
- 时间复杂度:O(n×(sum?target)),n为数组长度,sum是数组的和
- 空间复杂度:O(sum-target)
class Solution {
public int findTargetSumWays(int[] nums, int target) {
int sum = 0;
for (int num : nums) {
sum += num;
}
int diff = sum - target;
if (diff < 0 || diff % 2 != 0) {
return 0;
}
int n = nums.length, neg = diff / 2;
int[][] dp = new int[n + 1][neg + 1];
dp[0][0] = 1;
for (int i = 1; i <= n; i++) {
int num = nums[i - 1];
for (int j = 0; j <= neg; j++) {
dp[i][j] = dp[i - 1][j];
if (j >= num) {
dp[i][j] += dp[i - 1][j - num];
}
}
}
return dp[n][neg];
}
}