494. 目标和
难度:中等
给定一个非负整数数组,a1, a2, …, an, 和一个目标数,S。现在你有两个符号 +
和 -
。对于数组中的任意一个整数,你都可以从 +
或 -
中选择一个符号添加在前面。
返回可以使最终数组和为目标数 S 的所有添加符号的方法数。
示例:
输入:nums: [1, 1, 1, 1, 1], S: 3
输出:5
解释:
-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
一共有5种方法让最终目标和为3。
提示:
- 数组非空,且长度不会超过 20 。
- 初始的数组的和不会超过 1000 。
- 保证返回的最终结果能被 32 位整数存下。
解答:
class Solution {
//动态规划
//所有符号为+的元素和为x,符号为-的元素和的绝对值是y
//通过target = 正数和 - 负数和 = x - y,而x与y的和是数组总和:x + y = sum
//可以求出x = (target + sum) / 2
//转化为了求在数组中找出和为x的所有解决方案的问题
//时间复杂度O(N*target),空间复杂度O(target)
public int findTargetSumWays(int[] nums, int target) {
int sum = 0;
for(int num : nums) sum += num;
if((sum + target) % 2 == 1 || target > sum) return 0;
int x = (sum + target) / 2;
int[] dp = new int[x + 1];
dp[0] = 1;
for(int num : nums){
for(int j = x; j >= num; j--){
dp[j] += dp[j - num];
}
}
return dp[x];
}
}
参考自:
作者:LeetCode-Solution
链接:https://leetcode-cn.com/problems/target-sum/solution/494-mu-biao-he-dong-tai-gui-hua-zhi-01be-78ll/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。