leetcode 494

来源:互联网 发布:ubuntu 查看系统日志 编辑:程序博客网 时间:2024/06/06 03:21
该题目大意是给一个非负的整数数组,可以在每个元素前面加+或者-,给一个目标整数S,求问有多少种不同的添加方式,可以让数组中所有元素在添加符号之后相加的结果为S~

分析:深度优先搜索,尝试每次添加+或者-,当当前添加符号次数为nums数组的大小的时候,判断sum与S是否相等,如果相等就res++。sum为当前添加符号情况下的前面所有部分的总和~

class Solution {
public:
int findTargetSumWays(vector<int>& nums, int S) {
    int res = 0;
    dfs(nums, 0, S, 0, res);
    return res;
}


bool dfs(vector<int>& nums, int sum, int target, int pos, int& res){
    if(pos == nums.size()){
        if(sum == target) return true;
        return false;
    }
    sum += nums[pos];
    if(dfs(nums, sum, target, pos+1, res)) res++;
    sum -= nums[pos];
    
    sum -=nums[pos];
    if(dfs(nums, sum, target, pos+1, res)) res++;
    sum += nums[pos];
    
    return false;
}
};

0 0