LeetCode 494. 目标和
给你一个整数数组 nums 和一个整数 target 。
向数组中的每个整数前添加 ‘+’ 或 ‘-’ ,然后串联起所有整数,可以构造一个 表达式 :
例如,nums = [2, 1] ,可以在 2 之前添加 ‘+’ ,在 1 之前添加 ‘-’ ,然后串联起来得到表达式 “+2-1” 。
返回可以通过上述方法构造的、运算结果等于 target 的不同 表达式 的数目。
题解:
方法一:可以采用回溯的方式,这样时间复杂度为
O
(
2
n
)
O(2^n)
O(2n).
方法二:采用动态规划的方式。
假设数组中所有元素的和为sum,所有带符号的元素和的绝对值为neg,那么就有:
( s u m − n e g ) − n e g = t a r g e t (sum - neg) - neg = target (sum−neg)−neg=target
也就是 n e g = ( s u m − t a g e t ) / 2 neg = (sum - taget)/2 neg=(sum−taget)/2。 记 d i f f = s u m − t a r g e t 记diff = sum - target 记diff=sum−target
当diff为负数或者是奇数的时候,此题无解,应该返回0.
当diff为正偶数时,问题就变成,从n个数里选取c个数使得和为neg。
即为背包问题。状态转移方程为:
由于第 i 个状态只与 i - 1有关,那么可以用滚动数组优化。
class Solution {
private int ans = 0;
public int findTargetSumWays(int[] nums, int target) {
int n = nums.length,sum = 0;
for(int num : nums)
sum += num;
if(sum - target < 0 || (sum - target) % 2 != 0)return 0;
int[] dp = new int[(sum - target) / 2 + 1];
dp[0] = 1;
for(int num:nums)
for(int j = (sum - target) / 2; j >= num; j--){
dp[j] += dp[j - num];
}
return dp[(sum - target)/2];
}
}
题目来源:
https://leetcode-cn.com/problems/target-sum/
题解链接:https://leetcode-cn.com/problems/target-sum/solution/mu-biao-he-by-leetcode-solution-o0cp/
来源:力扣(LeetCode)