【ACWing】454. 表达式求值

题目地址：

https://www.acwing.com/problem/content/description/456/

给定一个只包含加法和乘法的算术表达式，请你编程计算表达式的值。

输入格式：
输入仅有一行，为需要你计算的表达式，表达式中只包含数字、加法运算符$+$和乘法运算符$\ast$，且没有括号，所有参与运算的数字均为$0$到$2^{31}-1$之间的整数。输入数据保证这一行只有$0\sim 9$、$+$、$\ast$这$12$种字符。

输出格式：
输出只有一行，包含一个整数，表示这个表达式的值。
注意：当答案长度多于$4$位时，请只输出最后$4$位，前导$0$不输出。

数据范围：
$0\le$表达式中加法运算符和乘法运算符的总数$\le 10^5$

可以用栈，用一个变量标记上次遇到的运算符是加号还是乘号，解析出一个数字之后，如果之前是加号，则直接将这个数字入栈，否则将栈顶乘以这个数字。最后对栈内数字求和。代码如下：

#include <iostream>
#include <stack>
using namespace std;

// 每个数最长可能是10位，所以字符串要开10^6级别的长度
const int N = 1e6 + 100, P = 1e4;
char s[N];
stack<long> stk;

int main() {
  scanf("%s", s);
  bool is_mult = false;
  for (int i = 0; s[i]; i++) {
    if (s[i] == '*') is_mult = true;
    else if (s[i] == '+') is_mult = false;
    else {
      int j = i;
      long num = 0;
      while (s[j] && '0' <= s[j] && s[j] <= '9') {
        num = (num * 10 + s[j] - '0') % P;
        j++;
      }
      i = j - 1;

      if (is_mult) {
        long top = stk.top();
        stk.pop();
        stk.push(top * num % P);
      } else stk.push(num);
    }
  }

  long res = 0;
  while (stk.size()) {
    res = (res + stk.top()) % P;
    stk.pop();
  }

  printf("%ld\n", res);
}

时空复杂度$O(n)$，$n$为字符串长度。