剑指 Offer II 102. 加减的目标值
给定一个正整数数组 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/YaVDxD
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
分析
方法一:空间复杂度O(nt)
n是数组长度,t是目标和。
预处理:令target = |target|,利用当前的target求解最终答案的结果跟之前的结果是一样的。假设添加“+”的数字和是p,添加“-”的数字和是q,那么满足p - q = taget的p和q一共有多少组答案就是多少;假设数组的总和是sum,那么p + q = sum ,两式相加得2 * p = sum + taget,这样,题目就转化为我们只要找到数组中有多少种不同的组合数字的和为p = (sum + target) / 2即可,如果sum + taget是奇数或者sum < taget说明没有满足条件的组合。
1、寻找状态转移方程。经过预处理后就跟上一题非常类似了,假设f(i, j)代表数组前 i 个数中和为 j 的不同组合数,那么当j > nums[i]时,f(i, j) = f(i - 1, j) + f(i - 1, j - nums[i])。当j = 0,f(i, j) = 1,当i = 0并且j > 0,f(i, j) = 0。
2、根据状态转移方程获取dp数组。先填充第一列,然后按照方程循环填充。
方法二:空间复杂度O(t)
内层循环用从右向左的遍历。
题解(Java)
方法一:空间复杂度O(nt)
class Solution {
public static int findTargetSumWays(int[] nums, int target) {
target = Math.abs(target);
int sum = 0;
for (int num : nums) {
sum += num;
}
if ((target + sum) % 2 == 1 || sum < target) {
return 0;
}
target = (target + sum) / 2;
int[][] dp = new int[nums.length + 1][target + 1];
dp[0][0] = 1;
for (int i = 1; i <= nums.length; i++) {
for (int j = 0; j <= target; j++) {
dp[i][j] = dp[i - 1][j];
if (j >= nums[i - 1]) {
dp[i][j] += dp[i - 1][j - nums[i - 1]];
}
}
}
return dp[nums.length][target];
}
}
方法二:空间复杂度O(t)
class Solution {
public static int findTargetSumWays(int[] nums, int target) {
target = Math.abs(target);
int sum = 0;
for (int num : nums) {
sum += num;
}
if ((target + sum) % 2 == 1 || sum < target) {
return 0;
}
target = (target + sum) / 2;
int[] dp = new int[target + 1];
dp[0] = 1;
for (int num : nums) {
for (int j = target; j >= num; j--) {
dp[j] += dp[j - num];
}
}
return dp[target];
}
}