AcWing 3302. 表达式求值

给定一个表达式，其中运算符仅包含 +,-,*,/（加 减 乘 整除），可能包含括号，请你求出表达式的最终值。

 

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

using namespace std;

stack<int> num;//树栈
stack<char> op;//符号栈

void eval()
{
    auto b = num.top(); num.pop();//b=第2个数。出栈
    auto a = num.top(); num.pop();//a=第1个树。出栈
    auto c = op.top(); op.pop();//c=符号。出栈
    int x;
    if (c == '+') x = a + b;
    else if (c == '-') x = a - b;
    else if (c == '*') x = a * b;
    else x = a / b;
    num.push(x);//计算结果入栈（数）
}

int main()
{
    unordered_map<char, int> pr{ {'+', 1}, {'-', 1}, {'*', 2}, {'/', 2} };//符号优先级。unordered_map是一个将key和value关联起来的容器，它可以高效的根据单个key值查找对应的value
    string str;
    cin >> str;
    for (int i = 0; i < str.size(); i++)    //对str逐个处理
    {
        auto c = str[i];    //c= str当前位
        if (isdigit(c))//isdigit()函数介绍：检查参数 c 是否为阿拉伯数字0 到9。
        {
            int x = 0, j = i;
            while (j < str.size() && isdigit(str[j]))//str.size()计算c++字符串长度。若是数字就循环
                x = x * 10 + str[j++] - '0';//将数加到x中。
            i = j - 1;//对i处理：上一行j++，这里要减回1。
            num.push(x);//将x入栈。
        }
        else if (c == '(') op.push(c);//符号入栈
        else if (c == ')')//对一对()进行处理
        {
            while (op.top() != '(') eval();//计算
            op.pop();//当前栈顶为（，出栈
        }
        else//c为符号的情况。
        {
            while (op.size() && op.top() != '(' && pr[op.top()] >= pr[c]) eval();//符号栈不为空，且栈顶不为（。‘pr[op.top()] >= pr[c]’判断优先级。计算栈中的符号
            op.push(c);//c入符号栈
        }
    }
    while (op.size()) eval();   //计算至符号栈为空。
    cout << num.top() << endl;  //输出结果
    return 0;
}