每日OJ题_DFS回溯剪枝⑧_力扣494. 目标和

目录

[力扣494. 目标和](#力扣494. 目标和)

解析代码(path设置成全局)

解析代码(path设置全局)


力扣494. 目标和

494. 目标和

难度 中等

给你一个非负整数数组 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
cpp 复制代码
class Solution {
public:
    int findTargetSumWays(vector<int>& nums, int target) {

    }
};

解析代码(path设置成全局)

在动态规划部分写过这题,现在用DFS写一写。对于每个数,可以选择加上或减去它,依次枚举每一个数字,在每个数都被选择时检查得到的和是否等于目标值。如果等于,则记录结果。

在这一题中可以窥见path是设置成全局的好还是设置成递归参数好:

如果path是一个类型,如 int 设置成递归参数好,参数可以帮我们完成回溯,代码会简洁。

如果path是一个数组之类的,设置成全局好,如力扣257. 二叉树的所有路径,否则效率低。

cpp 复制代码
class Solution {
    int path, ret, _target;
public:
    int findTargetSumWays(vector<int>& nums, int target) {
        _target = target;
        dfs(nums, 0);
        return ret;
    }

    void dfs(vector<int>& nums, int pos)
    {
        if(pos ==  nums.size())
        {
            if(path == _target)
                ++ret;
            return;
        }

        path += nums[pos]; // 加法
        dfs(nums, pos + 1);
        path -= nums[pos]; // 回溯

        path -= nums[pos]; // 减法
        dfs(nums, pos + 1);
        path += nums[pos]; // 回溯
    }
};

解析代码(path设置全局)

此题path设置全局效率并没有提高很多,但是代码会简洁:

cpp 复制代码
class Solution {
    int ret, _target;
public:
    int findTargetSumWays(vector<int>& nums, int target) {
        _target = target;
        dfs(nums, 0, 0);
        return ret;
    }

    void dfs(vector<int>& nums, int pos, int path)
    {
        if(pos ==  nums.size())
        {
            if(path == _target)
                ++ret;
            return;
        }
        dfs(nums, pos + 1, path + nums[pos]);  // 加法
        dfs(nums, pos + 1, path - nums[pos]); // 减法
    }
};
相关推荐
Trouvaille ~5 分钟前
【Linux】TCP Socket编程实战(一):API详解与单连接Echo Server
linux·运维·服务器·网络·c++·tcp/ip·socket
偷吃的耗子11 分钟前
【CNN算法理解】:CNN平移不变性详解:数学原理与实例
人工智能·算法·cnn
坚果派·白晓明17 分钟前
在鸿蒙设备上快速验证由lycium工具快速交叉编译的C/C++三方库
c语言·c++·harmonyos·鸿蒙·编程语言·openharmony·三方库
小镇敲码人24 分钟前
深入剖析华为CANN框架下的Ops-CV仓库:从入门到实战指南
c++·python·华为·cann
dazzle1 小时前
机器学习算法原理与实践-入门(三):使用数学方法实现KNN
人工智能·算法·机器学习
那个村的李富贵1 小时前
智能炼金术:CANN加速的新材料AI设计系统
人工智能·算法·aigc·cann
张张努力变强1 小时前
C++ STL string 类:常用接口 + auto + 范围 for全攻略,字符串操作效率拉满
开发语言·数据结构·c++·算法·stl
万岳科技系统开发1 小时前
食堂采购系统源码库存扣减算法与并发控制实现详解
java·前端·数据库·算法
小镇敲码人1 小时前
探索CANN框架中TBE仓库:张量加速引擎的优化之道
c++·华为·acl·cann·ops-nn
张登杰踩2 小时前
MCR ALS 多元曲线分辨算法详解
算法