Leetcode 241. 为运算表达式设计优先级 (递归分治）

最新推荐文章于 2022-09-08 22:01:05 发布

wwxy261

最新推荐文章于 2022-09-08 22:01:05 发布

阅读量167

点赞数

分类专栏：算法

本文链接：https://blog.csdn.net/wwxy1995/article/details/117170215

版权

算法专栏收录该内容

3633 篇文章 120 订阅

订阅专栏

这篇博客介绍了如何用优雅的Python和C++代码解决计算表达式的不同方式。通过递归地处理运算符，将表达式拆分成左右子问题并合并结果。在Python中，可以直接处理字符串，而在C++中需要额外处理多位数字的解析。这种方法展示了如何处理加法、减法和乘法运算符。

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

由于输出的表达式一定是合法的，且没有多余的字符串，因此我们对于任意一个运算符，可以分将这个运算符拆成左右两个运算符，递归解决左右两个子问题。解决完以后，然后合并成答案。

这里有两种方法，一种先做一个parse，将多位数字提前解析好，另一种不需要解析，直接对字符串进行处理，这种情况下Python实现起来更优雅。

优雅的Python代码

class Solution:
    def diffWaysToCompute(self, expression: str) -> List[int]:
        op = ['*', '+', '-']
        if expression.isdigit():
            return [int(expression)]
        res = []
        for i, c in  enumerate(expression):
            if c in op:
                left = self.diffWaysToCompute(expression[:i])
                right = self.diffWaysToCompute(expression[i+1:])
                for l in left:
                    for r in right:
                        if c == '*':
                            res.append(l * r)
                        elif c == '+':
                            res.append(l + r)
                        elif c == '-':
                            res.append(l - r)
        return res

下面是C++代码，由于C++ string不容易直接isdigit，因此需要这样写

class Solution {
public:
    vector<int> diffWaysToCompute(string expression) {
        vector<int> res;
        for(int i = 0; i < expression.size(); i++) {
            char c = expression[i];
            if (c == '*' || c == '-' || c == '+') {
                auto l = diffWaysToCompute(expression.substr(0, i));
                auto r = diffWaysToCompute(expression.substr(i + 1));
                for (auto v1 : l) {
                    for(auto v2 : r) {
                        switch(c) {
                            case '+' : res.push_back(v1 + v2); break;
                            case '-' : res.push_back(v1 - v2); break;
                            case '*' : res.push_back(v1 * v2); break;
                        }
                    }
                }
            }
        }
        if(res.size() == 0) {
            res.push_back(stoi(expression));
        }
        return res;
    }
};