Leetcode 494 目标和
给你一个整数数组 nums 和一个整数 target 。
向数组中的每个整数前添加 ‘+’ 或 ‘-’ ,然后串联起所有整数,可以构造一个 表达式 :
例如,nums = [2, 1] ,可以在 2 之前添加 ‘+’ ,在 1 之前添加 ‘-’ ,然后串联起来得到表达式 “+2-1” 。
返回可以通过上述方法构造的、运算结果等于 target 的不同 表达式 的数目。
示例 1:
输入:nums = [1,1,1,1,1], target = 3
输出:5
解释:一共有 5 种方法让最终目标和为 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
+1 + 1 + 1 + 1 - 1 = 3
思考过程
设目标和为s,nums数组的和为sum,sum可以分为x,y(x<y),即x+y=sum
因为目标和是s,所以x,y应满足y-x=s;
由此可以解除y=(sum+s)/2,这样题目就可以抓换为0-1背包问题,即用nums数组装满容量为y的背包有多少种方法
0-1 背包求解
①构造dp[i],表示容量为i的背包被装满有dp[i]种方法
②构造dp[i]的递推表达式,假设此时已经知道了dp[3]的解,需要求出dp[5]的解,那么需要nums数组有一个值为2,即nums[j]=2;
nums数组中有几个2,那么dp[5]就有几种被装满的方法,所以dp[5]+=dp[5-nums[j]];
推到可以知道dp[i]+=dp[i-nums[j]];
③初始化dp的值,dp[0]=1,即背包为空的情况下有一种装满方式
④需要二重循环来进行求解。
代码实现
class Solution {
public:
int findTargetSumWays(vector<int>& nums, int target) {
int y,sum=0;
for(auto i:nums) sum+=i;
if(abs(target)>sum) return 0;//注意此处要用一个绝对值,测试样例有一个卡
if((sum+target)%2==1) return 0;
y=(sum+target)/2;
vector<int> dp(y+1,0);
dp[0]=1;
for(int i=0;i<nums.size();++i){
for(int j=y;j>=nums[i];--j){
dp[j]+=dp[j-nums[i]];
}
}
return dp[y];
}
};