【Leetcode】227. Basic Calculator II

题目地址：

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

给出一个中缀表达式，求其值。表达式里只含数字（非负）、空格和加减乘除。

可以用两个栈，一个存数字，另一个存运算符。保持运算符栈单调，从栈底到栈顶运算符优先级升高（或相等）。规定乘除优先级高于加减，乘除优先级一样，加减优先级也一样。接下来遍历字符串，遇到空格直接略过；遇到数字则解析出操作数，存入栈中；遇到运算符，则看一下该运算符和运算符栈顶优先级哪个高，如果栈顶高，则出栈并与操作数栈中最上面两个数字做运算，结果再入栈。遍历完字符串后如果操作数栈里还有不止一个数字，说明还有运算没算完，则继续算，算到只剩最后一个数字为止。代码如下：

class Solution {
 public:
  int calculate(string s) {
    auto f = [](int x, int y, char op) {
      switch (op) {
        case '+': return x + y;
        case '-': return x - y;
        case '*': return x * y;
        case '/': return x / y;
      }

      return 0;
    };
    stack<int> stk;
    stack<char> ops;
    unordered_map<char, int> mp = {{'+', 1}, {'-', 1}, {'*', 2}, {'/', 2}};
    for (int i = 0; i < s.size(); i++) {
      if (s[i] == ' ') continue;
      if (isdigit(s[i])) {
        int x = 0;
        int j = i;
        while (j < s.size() && isdigit(s[j])) x = x * 10 + (s[j++] - '0');
        i = j - 1;
        stk.push(x);
      } else {
        while (ops.size() && mp[ops.top()] >= mp[s[i]]) {
          char op = ops.top(); ops.pop();
          int n2 = stk.top(); stk.pop();
          int n1 = stk.top(); stk.pop();
          stk.push(f(n1, n2, op));
        }
        ops.push(s[i]);
      }
    }

    while (stk.size() > 1) {
      char op = ops.top(); ops.pop();
      int n2 = stk.top(); stk.pop();
      int n1 = stk.top(); stk.pop();
      stk.push(f(n1, n2, op));
    }

    return stk.top();
  }
};

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

上面的方法还是麻烦了些，我们可以直接通过维护一个状态来做。整数关于加法和乘法形成一个环。初始状态为$(0,1)$（分别是加法单位元和乘法单位元，这个项的含义是$0+1\#$，$\#$的意思是下面的运算符还未定义），接着我们遍历表达式，遍历的时候得到每个项和其前面的运算符。最开始的项的运算符定义为乘法。设当前的状态为$(a,b)$，如果下一项的运算符是乘除法，则说明前面遍历完留的末尾数需要参与乘除运算，例如如果下一项是$x$并且是乘法，则状态变为$(a,b\times x)$，如果是除法则为$(a,b/x)$；如果下一项的运算符是加减法，则说明前面遍历完的项可以合并了，如果是加法，则状态变为$(a+b,x)$，否则变为$(a+b,-x)$。最后遍历完之后，将两维相加即得答案。代码如下：

class Solution {
 public:
  int calculate(string s) {
    int a = 0, b = 1;
    char op = '*';
    for (int i = 0; i < s.size(); i++) {
      if (s[i] == ' ') continue;
      if (isdigit(s[i])) {
        int j = i;
        int x = 0;
        while (j < s.size() && isdigit(s[j])) x = x * 10 + (s[j++] - '0');
        i = j - 1;
        if (op == '+') a += b, b = x;
        else if (op == '-') a += b, b = -x;
        else if (op == '*') b *= x;
        else b /= x;
      } else op = s[i];
    }

    return a + b;
  }
};

时间复杂度$O(n)$，空间$O(1)$。