You are given a list of non-negative integers, a1, a2, …, an, and a target, S. Now you have 2 symbols + and -. For each integer, you should choose one from + and - as its new symbol.
Find out how many ways to assign symbols to make sum of integers equal to target S.
Example 1:
Input: nums is [1, 1, 1, 1, 1], S is 3.
Output: 5
Explanation:
-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
给出一个数组nums和一个目标和S,可以用+,-给数组中每个数字做运算符,问有多少种方法可以得到目标S
思路:
定义DP数组,dp[i][j]表示nums到下标i 为止的部分达到目标和j 有多少种方法。
定义nums的长度为n,那么结果只需要返回dp[n][S]
而所有可能的和的上限是数组nums中所有数字相加,下限是所有的数都取减号。
一个数字都不取的时候,即dp[0][-sum~sum] = 0
取一个数字的时候,dp[1][-nums[1]] = dp[1][nums[1]] = 1
同理取数字nums[i], 和为j时,上一步的和可以是j - nums[i], 也可以是j + nums[i], 因此dp[i][j] = dp[i-1][j - nums[i]] + dp[i-1][j + nums[i]]
因为数组下标不能为负值,因此负值的部分需要映射
即-sum~sum需要映射到0~2*sum
最后的和S也要映射到S+sum
public int findTargetSumWays(int[] nums, int S) {
if(nums == null || nums.length == 0) {
return 0;
}
int sum = 0;
for(int i = 0; i < nums.length; i++) {
sum += nums[i];
}
int[][] dp = new int[nums.length + 1][2*sum + 1];
for(int i = 1; i < nums.length; i++) {
for(int j = 0; j < sum; j++) {
dp[i][j] = dp[i - 1][j - nums[i - 1] + sum] +
dp[i - 1][j + nums[i - 1] + sum];
}
}
return dp[nums.length - 1][S + sum];
}