[AcWing 3302] 表达式求值

---

点击查看代码

#include<iostream>
#include<stack>
#include<cstring>
#include<unordered_map>

using namespace std;

stack<int> nums;
stack<char> op;
unordered_map<char, int> h{ {'+', 1}, {'-', 1}, {'*', 2}, {'/', 2} };
void eval()
{
    int b = nums.top();   nums.pop();
    int a = nums.top();   nums.pop();
    char p = op.top();  op.pop();
    int r = 0;
    if (p == '+')   r = a + b;
    if (p == '-')   r = a - b;
    if (p == '*')   r = a * b;
    if (p == '/')   r = a / b;
    nums.push(r);
}
int main()
{
    string s;
    cin >> s;
    for (int i = 0; i < s.size(); i ++) {
        if (isdigit(s[i])) {
            int x = 0, j = i;
            while (j < s.size() && isdigit(s[j])) {
                x = x * 10 + s[j] - '0';
                j ++;
            }
            nums.push(x);
            i = j - 1;
        }
        else if (s[i] == '(')   op.push(s[i]);
        else if (s[i] == ')') {
            while (op.top() != '(') eval();
            op.pop();
        }
        else {
            while (op.size() && h[op.top()] >= h[s[i]]) eval();
            op.push(s[i]);
        }
    }
    while (op.size())   eval();
    cout << nums.top();
    return 0;
}

---

1. 用字符串进行读入，对每一位字符进行如下操作：
   ① 如果当前位置是数字，使用 while 获取所有的数字部分，直到遇到负号，用 x 存下数字，push 到 nums 中，然后 i = j - 1（需要 -1 是因为 for 循环会执行 i ++）；
   ② 如果当前位置是 '('，把 '(' 放到 op 中；
   ③ 如果当前位置是 ')'，则从 op 中弹出操作符，直到遇到 '('，每次使用 eval 弹出操作符和 2 个数字，并计算当前操作符和 2 个数字的运算结果 r，将 r 放到 nums 中；
   ④ 当前位置 s[ i ] 是字符且不为 '(' ' )'， op 栈顶元素的优先级比 s[ i ] 的优先级要高，则需要使用 eval 弹出比 s[ i ] 优先级高的操作符进行运算，最后把 s[ i ] 放到栈顶；
2. 如果最后 op 不为空，则需要把 op 里面的操作符都弹出来进行运算；
3. 最终的结果会存在 nums 中；
4. eval 操作的注意事项：
   ① nums 中弹出数时，先弹出的是 b，后弹出的是 a（ a - b，a / b ）；
   ② 从 nums 和 op 这两个栈中弹出元素时，先 top( ) 取出栈顶元素，再用 pop( ) 弹出；
5. 使用 unordered_map<char, int> h 来维护操作符的优先级；