494.给你一个整数数组 nums 和一个整数 target 。
向数组中的每个整数前添加 ‘+’ 或 ‘-’ ,然后串联起所有整数,可以构造一个 表达式 :
例如,nums = [2, 1] ,可以在 2 之前添加 ‘+’ ,在 1 之前添加 ‘-’ ,然后串联起来得到表达式 “+2-1” 。
返回可以通过上述方法构造的、运算结果等于 target 的不同 表达式 的数目。
- 该题显然可以用回溯的方法来解决,整数数组的每个元素最终变成表达式后只有两种选择,利用回溯找到所有的表达式,当表达式和等于target的时候计数器加一即可。
- 另外一种解决方法是将题目转化成背包问题来进行求解。由于整数数组的元素都是非负整数,则设整数数组的和为sum,添加了+号的元素的和为x,则添加-号都元素的和为sum-x,那么我们要找的情况就是当x-(sum-x)=target的情况,即x=(sum+target)/2的情况,则该题目就转化成了求背包容量为x时,从数组中找元素填满背包有多少种情况的组合问题。需要注意的是当target的绝对值大于sum时以及当sum+target不为偶数时(因为x必须时非负整数,所以sum+target必须为偶数)都构造不出表达式使得和为target。
- 那么接下来就是确定dp数组的含义、初始化及其递推公式,这里用一维的dp数组,dp[j]代表填满容量为j的背包有dp[j]种方法。初始化dp[0]为1,因为容量为0的背包填满只有一种情况,那就是不放物品,其余元素都初始化为0。当确定了背包的其中一个元素nums[i]的时候,那么dp[j] = dp[j - nums[i]] + dp[j],所以递推公式应该是dp[j] += dp[j - nums[i]]。
- 遍历顺序跟01背包问题的遍历顺序是一样的,两层循环嵌套,nums放在外层循环,内层循环不断更新dp数组。最后返回dp[bagSize]。
- 代码如下:
class Solution {
public int findTargetSumWays(int[] nums, int target) {
int sum = 0;
for (int i = 0; i < nums.length; i++) {
sum += nums[i];
}
if (Math.abs(target) > sum)
return 0;
if ((target + sum) % 2 == 1)
return 0;
int bagSize = (target + sum) / 2;
int[] dp = new int[bagSize + 1];
dp[0] = 1;
for (int i = 0; i < nums.length; i++) {
for (int j = bagSize; j >= nums[i]; j--) {
dp[j] += dp[j - nums[i]];
}
}
return dp[bagSize];
}
}