【Leetcode】772. Basic Calculator III

题目地址：

https://leetcode.com/problems/basic-calculator-iii/

给定一个中缀表达式，以字符串$s$形式给出，包含空格，数字（题目保证除非是$0$，否则不以$0$开头），加减乘除和左右小括号。要求求其值。

经典算法是用两个栈，一个存操作数，另一个存操作符。具体算法可以参考https://blog.csdn.net/qq_46105170/article/details/106209367。大致思路如下：
1、先规定每个运算符（还有左括号）的优先级，左括号优先级最小，为$1$，然后加减优先级都是$2$，接着乘除优先级都是$3$；
2、接着遍历$s$（下面不考虑遇到空格的情形），遇到左括号，则入操作符栈；
3、遇到数字，则入操作数栈；
4、遇到操作符，则先看一下操作符栈的栈顶的优先级是否大于等于该操作符，如果确实大于等于，说明栈顶这个运算可以算了，从操作数栈里pop出两个数，进行计算，然后再push回操作数栈，如此这般，直到操作符栈的栈顶的优先级是小于新来的操作符的，再将该操作符入栈；这里的关键是维护操作符栈按照优先级是严格从小到大的即可；
5、遇到右括号，则只要操作符栈的栈顶不是左括号，就不停对操作符栈顶进行计算，每次都从操作数栈pop两个数，算完了push回去，再pop两个数，算完了再push回去，直到操作符栈顶是左括号为止，再将左括号出栈。
$s$遍历完之后，操作数栈里如果还有多于一个数，就继续进行计算，直到只有一个数了，它就是结果。

由于要防止溢出，我们干脆用long来计算。代码如下：

class Solution {
 public:
  int calculate(string s) {
    unordered_map<char, int> mp{
        {'(', 1}, {'+', 2}, {'-', 2}, {'*', 3}, {'/', 3}};
    stack<char> ops;
    stack<long> stk;
    auto calc = [&](stack<long>& stk, stack<char>& ops) {
      long b = stk.top(); stk.pop();
      long a = stk.top(); stk.pop();
      switch (ops.top()) {
        case '+': stk.push(a + b); break;
        case '-': stk.push(a - b); break;
        case '*': stk.push(a * b); break;
        case '/': stk.push(a / b); break;
      }
      ops.pop();
    };
    for (int i = 0; i < s.size(); i++) {
      char ch = s[i];
      if (ch == '(') ops.push('(');
      else if (isdigit(ch)) {
        int j = i;
        while (j < s.size() && isdigit(s[j])) j++;
        stk.push(stol(s.substr(i, j - i)));
        i = j - 1;
      } else if (ch == ')') {
        while (ops.top() != '(') calc(stk, ops);
        ops.pop();
      } else if (ch != ' ') {
        while (ops.size() && mp[ops.top()] >= mp[ch])
          calc(stk, ops);
        ops.push(ch);
      }
    }

    while (stk.size() > 1) calc(stk, ops);
    return (int)stk.top();
  }
};

时空复杂度$O(n)$。