LeetCode 494.目标和【C++】

最新推荐文章于 2024-06-12 10:00:23 发布

G.X.Y~苏

最新推荐文章于 2024-06-12 10:00:23 发布

阅读量120

点赞数

分类专栏： LeetCode 文章标签： leetcode c++ 算法

本文链接：https://blog.csdn.net/qq_52111135/article/details/129717717

版权

LeetCode 专栏收录该内容

22 篇文章 1 订阅

订阅专栏

一、题目

给你一个整数数组 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

示例 2：

输入： nums = [1], target = 1
输出： 1

提示：

1 <= nums.length <= 20
0 <= nums[i] <= 1000
0 <= sum(nums[i]) <= 1000
-1000 <= target <= 1000

二、解题思路

本题可看作0-1背包问题。
设：加法得到的总和 x ，减法得到的总和 y ，则数组总和 sum=x+y ; target=x-y ;
由此可推算出 sum = x + ( x - target ) => x = ( sum + target ) / 2
因此，求出使总和为x 有几种方法就行。可以看做 装满容量为 x 的背包有几种方法。
则

dp[j] 可看作装满容量为 j 的背包，有 dp[j] 种方法
若j=5, 求装满容量为5的背包有几种方法（一定要记住 j 和 dp[j] 代表的意思，结合下面的例子理解）

有个 1（nums[i]=1）=> 有dp[4]种方法凑容量为5的背包
（此时背包容量为4，恰好有个数字 1 凑成容量5，凑容量4有 dp[4] 种方法，则此时有 dp[4] 种方法凑容量5）
有个 2（nums[i]=2）=> 有dp[3]种方法凑容量为5的背包
有个 3（nums[i]=3）=> 有dp[2]种方法凑容量为5的背包
有个 4（nums[i]=4）=> 有dp[1]种方法凑容量为5的背包
有个 5（nums[i]=5）=> 有dp[0]种方法凑容量为5的背包

因此凑容量为5的背包的方法共有 dp[5]= dp[4]+ dp[3] + dp[2] + dp[1] + dp[0]
可推导出式子：dp[j] += dp[ j-nums[i] ]
3. 由第二步的推导可以看出，dp[0] = 1,其余初始化为0,因为其他都是从 dp[0] 开始算出来。

三、代码

class Solution {
public:
    int findTargetSumWays(vector<int>& nums, int target) {
        
        int sum=0;
        for(int i=0;i<nums.size();i++) sum+=nums[i];
        if(abs(target)>sum) return 0;
        if((target+sum)%2==1) return 0;
        int temp=(target+sum)/2;
        vector<int> dp(temp+5,0);
        dp[0]=1;
        for(int i=0;i<nums.size();i++){
            for(int j=temp;j>=nums[i];j--){
                dp[j]+=dp[j-nums[i]];
            }
        }
        return dp[temp];
    }
};