Leetcode 282. 给表达式添加运算符 C++

Leetcode 282. 给表达式添加运算符

题目

给定一个仅包含数字 0-9 的字符串和一个目标值，在数字之间添加二元运算符（不是一元）+、- 或 * ，返回所有能够得到目标值的表达式。

测试样例

示例 1:

输入: num = "123", target = 6
输出: ["1+2+3", "1*2*3"] 

示例 2:

输入: num = "232", target = 8
输出: ["2*3+2", "2+3*2"]

示例 3:

输入: num = "105", target = 5
输出: ["1*0+5","10-5"]

示例 4:

输入: num = "00", target = 0
输出: ["0+0", "0-0", "0*0"]

示例 5:

输入: num = "3456237490", target = 9191
输出: []

题解

dfs深搜回溯
我们选数时只需要考虑有没有前导零，要避免前导零。然后我们再考虑三种情况，+、-、*，对于+、-直接运算更新结果即可；但 ，需要特别处理一下，因为它的优先级要比+、-要高，和是相同优先级，因此我们不妨记录一下上次的操作，在进行 * 运算时，先把上次的操作去掉，更新操作即可。详细过程见代码

代码

vector<string> ans;
    void dfs(int idx,long now,int pre,string& nowStr,string& str,int len,int target){
        if(idx == len){
            if(now == target){
                ans.push_back(nowStr);
            }
                
        }else if(idx == 0){		//第一次选数时，我们只需要选数即可，不需要运算
            for(int i=idx; i<len; i++){
                if(i!=idx && nowStr[0]=='0')    return;		//避免前导零
                nowStr += str[i]; 
                now = now*10 + str[i]-'0';
                dfs(i+1,now,now,nowStr,str,len,target);
            }
        }else{
            long cur = 0;
            string curStr;
            int oldStrLen = nowStr.length();
            for(int i=idx; i<len; i++){
                if(i!=idx && curStr[0]=='0')    return;		//避免前导零
                cur = cur*10 + str[i]-'0';
                curStr += str[i];
                nowStr += "+" + curStr;
                dfs(i+1,now+cur,cur,nowStr,str,len,target);		//加运算
                nowStr.resize(oldStrLen);
                nowStr += "-" + curStr;
                dfs(i+1,now-cur,-cur,nowStr,str,len,target);	//减运算
                nowStr.resize(oldStrLen);
                nowStr +=  "*" + curStr;
                dfs(i+1,now-pre+pre*cur,pre*cur,nowStr,str,len,target);		//乘运算，先把上次的操作减掉
                nowStr.resize(oldStrLen);
            }
        }
    }
    vector<string> addOperators(string num, int target) {
        int len = num.length();
        string now;
        dfs(0,0,0,now,num,len,target);
        return ans;
    }

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/expression-add-operators
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。