Leetcode | Expression Add Operators

来源:互联网 发布:发散和收敛定义知乎 编辑:程序博客网 时间:2024/05/29 12:40

原题链接:https://leetcode.com/problems/expression-add-operators

题目内容如下所示:

Given a string that contains only digits 0-9 and a target value, return all possibilities to add binary operators (not unary) +, -, or * between the digits so they evaluate to the target value.

Examples:
“123”, 6 -> [“1+2+3”, “1*2*3”]
“232”, 8 -> [“2*3+2”,”2+3*2”]
“105”, 5 -> [“1*0+5”,”10-5”]
“00”, 0 -> [“0+0”, “0-0”, “0*0”]
“3456237490”, 9191 -> []

题目的大概意思就是给出一个包含0-9的字符串,然后向其中任意添加一些二元操作符,包括+、-和*,添加后会得到一个计算表达式,算出来的结果要是后面给出的数字target,所以是要我们输出所有可能的表达式的集合。

这道题刚着手时并不简单,虽然题意看起来很容易理解,但仔细想想其实还挺复杂的,因为它可以是相邻数字组在一起当成一个更大的数字,而不是固定地在字符串中每个数字中间插一个操作符,所以这其实可以看成一个动态规划的问题,我们可以先从它的子问题入手,在这道题就相当于是子字符串,然后找出子串和原串之间的联系,这里其实可以用dfs的思想,通过递归遍历的方式逐渐找出问题的所有解。

最后给出代码如下所示:

class Solution {private:    void dfs(std::vector<string>& res, const string& num, const int target, string cur, int pos, const long cv, const long pv, const char op) {        if (pos == num.size() && cv == target) {            res.push_back(cur);        } else {            for (int i=pos+1; i<=num.size(); i++) {                string t = num.substr(pos, i-pos);                long now = stol(t);                if (to_string(now).size() != t.size()) continue;                dfs(res, num, target, cur+'+'+t, i, cv+now, now, '+');                dfs(res, num, target, cur+'-'+t, i, cv-now, now, '-');                dfs(res, num, target, cur+'*'+t, i, (op == '-') ? cv+pv - pv*now : ((op == '+') ? cv-pv + pv*now : pv*now), pv*now, op);            }        }    }public:    vector<string> addOperators(string num, int target) {        vector<string> res;        if (num.empty()) return res;        for (int i=1; i<=num.size(); i++) {            string s = num.substr(0, i);            long cur = stol(s);            if (to_string(cur).size() != s.size()) continue;            dfs(res, num, target, s, i, cur, cur, '#');                }        return res;    }};
原创粉丝点击