Leetcode 494. 目标和 C++

Leetcode 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 位整数存下。

题解

动态规划
对于数组中的数，我们分为两组，一组为正数，一组为负数，也就是说找在集合中找一些数T作为正数，那么剩下的数N就是负数，然后让S(T)-S(N) = S，即S(T) = S + S(N)。我们等号两边同时加上S(T)，得到2*S(T) = S +sum，进而求得S(T) = (S+sum)/2。对于正数的选择，就是最终答案的选择
我们令dp[i]表示和为i的集合个数
对于第i个数的选择，则dp[j] += dp[j-nums[i]]，其中j>=nums[i]。最终我们返回dp[(S+sum)/2]
详细过程见代码

代码

	int findTargetSumWays(vector<int>& nums, int S) {
        int n = nums.size();
        int sum = 0;
        for(int i=0; i<n; i++)
            sum += nums[i];
        if(sum<S || (S+sum) % 2 != 0)  return 0;
        S = (S+sum)/2;
        vector<int> dp(S+1,0);
        dp[0] = 1;
        for(int i=0; i<n; i++){
            for(int j=S; j>=nums[i]; j--){
                dp[j] += dp[j-nums[i]]; 
            }
        }
        return dp[S];
    }

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/target-sum
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。