[Leetcode] 772. Basic Calculator III 解题报告

LeetCode 772: 基本计算器III解题解析

最新推荐文章于 2024-08-09 22:43:08 发布

原创最新推荐文章于 2024-08-09 22:43:08 发布 · 4.4k 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#Leetcode #解题报告 #string #recursion

IT公司面试习题专栏收录该内容

736 篇文章

订阅专栏

实现一个基本计算器，评估简单表达式字符串。表达式包含非负整数、加减运算符、括号。遵循非负整数除法向零取整的规则。给定的表达式总是有效的。本文介绍了解题思路和代码实现，重点在于递归处理括号内的计算，并使用“引用传递”技巧处理全局变量。

题目：

Implement a basic calculator to evaluate a simple expression string.

The expression string may contain open ( and closing parentheses ), the plus + or minus sign -, non-negative integers and empty spaces .

The expression string contains only non-negative integers, +, -, *, / operators , open ( and closing parentheses ) and empty spaces . The integer division should truncate toward zero.

You may assume that the given expression is always valid. All intermediate results will be in the range of [-2147483648, 2147483647].

Some examples:

"1 + 1" = 2
" 6-4 / 2 " = 4
"2*(5+5*2)/3+(6/2+8)" = 21
"(2+6* 3+5- (3*14/7+2)*5)+3"=-12

Note: Do not use the eval built-in library function.

思路：

我们顺序循环，处理字符串中的各个字符：如果遇到‘+’或者‘-’，就将其加入到数组nums中；如果遇到‘*’或者‘/’，则直接进行计算；如果遇到'('，则递归调用parseExpr函数，返回括号对里面的计算结果。当所有的数都计算完毕之后，将数组nums中的所有数加起来，返回和就可以了。

这里的奥妙在于采用"pass by reference"的方法，使得i成为“全局变量”；另外就是合理地设计递归调用的函数，以处理括号嵌套的问题。

代码：

class Solution {
public:
    int calculate(string s) {
        int i = 0;
        return parseExpr(s, i);
    }
private:
    int parseExpr(string &s, int &i) {  // parse the expression
        vector<int> nums;
        char op = '+';
        int n;
        for (; i < s.length() && op != ')'; ++i) {
            if (s[i] == ' ') {
                continue;
            }
            if (s[i] == '(') {
                n = parseExpr(s, ++i);
            }
            else {
                n = parseNum(s, i);
            }
            switch(op) {
                case '+':
                    nums.push_back(n);
                    break;
                case '-':
                    nums.push_back(-n);
                    break;
                case '*':
                    nums.back() *= n;
                    break;
                case '/':
                    nums.back() /= n;
                    break;
            }
            op = s[i];
        }
        int res = 0;
        for (int n : nums) {
            res += n;
        }
        return res;
    }
    
    int parseNum(string &s, int &i) {       // parse the num
        int n = 0;
        while (i < s.length() && isdigit(s[i])) {
            n = 10 * n + s[i++] - '0';
        }
        return n;
    }
};