LeetCode 494. Target Sume

题解

之前在luogu上做过一道类似的题，但是更难点还要考虑空格。记得也写过题解的但是找不到了。。。
这道简化了点，所以我用dp来做一下，思路比较清晰。
dp[ i , j ] : 表示前 i 个数组合成总和 j 的方案数
有转移方程：
dp[ i , j ] = dp[ i-1 , j - nums[i] ] + dp[ i-1 , j + nums[i] ]
这很显然吧。

注意j不能为负数，所以需要一个偏移量bar使之为正。
还要注意边界。。。空间还可以优化。。。
还有更优解。。。

---

Code

class Solution {
public:
    int dp[20][4002];
    const static int bar = 2000;
    int findTargetSumWays(vector<int>& nums, int S) {
        memset(dp,0,sizeof(dp));
        
        if(nums[0]==0) dp[0][bar]=2;// keng 1
        else dp[0][bar+nums[0]]=dp[0][bar-nums[0]]=1;
        
        int n = nums.size();
        if(S>1000 || S<-1000) return 0; // keng 2
        if(n==1){
            return dp[0][S+bar];
        }
        for(int i=1;i<n;i++){
            for(int j=-1000;j<=1000;j++){
                dp[i][j+bar] = dp[i-1][j+bar +nums[i]]+dp[i-1][j+bar -nums[i]];
            }
        }
        return dp[n-1][S+bar];
    }
};