i007.vip

i007.vip

优先队列-降维打击

04.算法训练

优雅的枚举(回溯)

给你一个整数数组 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
相关标签
数组
动态规划
回溯

C++

 

作者:力扣 (LeetCode)
链接:https://leetcode-cn.com/leetbook/read/queue-stack/ga4o2/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。

 

class Solution {
public:
    int findTargetSumWays(vector<int>& nums, int target) {
        if (nums.size() == 0)
            return 0;

        int result = 0;
        backtrace(nums, target, result, 0, 0);
        return result;
    }

    void backtrace(const vector<int>& nums, int target, int& result, int level, int sum) {
        if (level == nums.size()) {
            if (sum == target) {
                result++;
            }
            return;
        }

        backtrace(nums, target, result, level + 1, sum + nums[level]);
        backtrace(nums, target, result, level + 1, sum - nums[level]);
    }
};

 

发表回复