目标和
题目:
给你一个整数数组 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
class Solution {
public int findTargetSumWays(int[] nums, int target) {
int sum = 0;
// 求数组元素的绝对和, 来确定数组的边界
for(int num : nums) {
sum += Math.abs(num);
}
if(target > sum) {
return 0;
}
// dp[i][j]表示 前 i 个数的和为 j 的表达式数目
// 由于数组下标从0开始 那么dp[i][0] 表示的是前 i 个数的和为 -sum
int dp[][] = new int[nums.length + 1][2 * sum + 1];
// 初始化前 0 个数的和为 0的表达式数目为 1
dp[0][0 + sum] = 1;
for(int i = 1; i <= nums.length; i++) {
int num = nums[i - 1];
for(int j = -sum; j <= sum; j++) {
// 动态规划方程:
// dp[i][j] = dp[i - 1][j - nums[i - 1]] + dp[i - 1][j + nums[i - 1]]
if(j - num + sum >= 0) { // 防止下标越界
dp[i][j + sum] += dp[i - 1][j + sum - num];
}
if(j + num + sum <= sum * 2) {
dp[i][j + sum] += dp[i - 1][j + sum + num];
}
}
}
return dp[nums.length][target + sum];
}
}