Leetcode - 494 Target Sum (Medium)
题目描述:给定一个正整数数组,添加正负号使得最终和为 target,求共有几种添加正负号的方式。
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
There are 5 ways to assign symbols to make the sum of nums be target 3.
解题思路:假定 P 为正号集合,N 为负号集合,所有数字的和为 sum,目标值为 target。
比如给定 nums = [1, 2, 3, 4, 5]
和target = 3
,则 +1-2+3-4+5 = 3
为一个可能的解法,其中 P = [1, 3, 5]
、N = [2, 4]
、sum = 15
。
S(P) + S(N) = sum
S(P) - S(N) = target
S(P) * 2 = sum + target
题目转换成找到满足 positive = (sum + target) / 2 的子集。
public int findTargetSumWays(int[] nums, int S) {
int sum = 0;
for(int num : nums){
sum += num;
}
if (sum < S || (sum + S) % 2 == 1) {
return 0;
}
int W = (sum + S) / 2;
int[] dp = new int[W + 1];
dp[0] = 1;
for (int num : nums) {
for (int i = W; i >= num; i--) {
dp[i] = dp[i] + dp[i - num];
}
}
return dp[W];
}