力扣 282. 给表达式添加运算符 dfs 回溯

最新推荐文章于 2022-09-11 23:05:17 发布

csu_xiji

最新推荐文章于 2022-09-11 23:05:17 发布

阅读量173

点赞数

分类专栏：力扣 DFS 回溯文章标签：算法 c++ leetcode

本文链接：https://blog.csdn.net/xiji333/article/details/120806563

版权

力扣同时被 3 个专栏收录

282 篇文章 2 订阅

订阅专栏

DFS

62 篇文章 0 订阅

订阅专栏

回溯

18 篇文章 0 订阅

订阅专栏

该博客介绍了如何运用回溯算法解决LeetCode中的一个问题——给定一个仅包含数字的字符串，添加运算符（+，-，*）使其结果等于目标值。博主分享了C++实现的代码，并指出在处理过程中需要注意数字0作为起始数字时不能与其他数字连接的细节。

摘要由CSDN通过智能技术生成

https://leetcode-cn.com/problems/expression-add-operators/

思路：很明显的回溯了吧。。考虑每个位置的运算符种类有三种以及数字可以连起来，那么相当于每个位置有四种选择，直接暴力计算时间复杂度为 $O(4^{n})$ ，考虑 $n$ 最大不过等于10，还是可以通过滴~需要注意的一个细节问题是，如果以位置 $i$ 为一个数的起始，且其为0，那么后续就不能再跟其他数字了！

class Solution {
public:
    vector<string> addOperators(string num, int target) {
        vector<string> ans;
        using ll=long long;
        string expr;
        function<void(int, ll, ll)> cal=[&](int pos, ll val, ll pre)
        {
            int n=num.size();
            if(pos>=n)
            {
                if(val==target)
                    ans.push_back(expr);
                return;
            }
            int curSize=expr.size();
            if(pos>0)
                expr+=' ';
            ll curVal=0;
            for(int i=pos;i<n;i++)
            {
                int posVal=num[i]-'0';
                curVal=curVal*10+posVal;
                expr+=num[i];
                if(pos==0)
                    cal(i+1, curVal, curVal);
                else
                {
                    expr[curSize]='+';
                    cal(i+1, val+curVal, curVal);
                    expr[curSize]='-';
                    cal(i+1, val-curVal, -curVal);
                    expr[curSize]='*';
                    cal(i+1, val-pre+curVal*pre, curVal*pre);
                }
                if(i==pos&&num[pos]=='0')
                    break;
            }
            expr.resize(curSize);
        };
        cal(0, 0, 0);
        return ans;
    }
};