目标和(leetcode)

问题描述

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

代码

//类似子集树，但没有剪枝函数
//思路是抽象成二叉树，左孩子是加  右孩子是减，然后递归求解  遍历所有节点

class Solution {
public:
    int findTargetSumWays( vector<int>& nums, int target ) {
        int answer = 0;
        int numSize = nums.size();
        int t = 0, sum = 0;//t为树的层数  sum为当前层数以即前面数之和
        backTrack( t, target, sum, answer, nums, numSize );
        return answer;

    }

    void backTrack( int t, int target, int sum, int& answer, vector<int>& nums, int numSize ) {
        //到达叶子节点
        if( t >= numSize ) {
            //如果找到一种解决方法  答案+1
            if( sum == target ) {
                answer++;
            }

            return;
        }

        backTrack( t + 1, target, sum + nums[t], answer, nums, numSize );	//遍历左子树
        backTrack( t + 1, target, sum - nums[t], answer, nums, numSize );	//遍历右子树
    }
};