表达式求值(栈+unordered_map)

第一次写博客，见谅hh

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

Input: (2+2) * (1+1）

Output: 8

原题链接

# include <bits/stdc++.h>

using namespace std;

stack<int> num;
stack<char> op;

void eval()//从后往左取值计算
{
    auto b = num.top(); num.pop();
    auto a = num.top(); num.pop();
    auto c = op.top(); op.pop();
    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先预处理各运算符的优先级
    string str;
    cin >> str;
    for (int i = 0; i < str.size(); i ++ )
    {
        auto c = str[i];
        if (isdigit(c))//这是一种很好的查找数字的方法
        {
            int x = 0, j = i;
            while (j < str.size() && isdigit(str[j]))
                x = x * 10 + str[j ++ ] - '0';
            i = j - 1;//因为上面是j++
            num.push(x);//把获取的数字入栈
        }
        else if (c == '(') op.push(c);
        else if (c == ')')//遇到反括号，我们就可以优先算出整个括号里面的了
        {
            while (op.top() != '(') eval();
            op.pop();//把'('弹出
        }
        else//这种情况举例有(2*3+2)、2*3+2
        {//即c是运算符，且在它之已经有一个运算符了(+之前已经有*了)
            while (op.size() && op.top() != '(' && pr[op.top()] >= pr[c]) eval();
            //如果当前运算符比之前运算符优先级要小，
            //那么我们可以先算出之前的表达式(因为优先算)再把c插入
            op.push(c);
        }
    }
    while (op.size()) eval();//最后整个计算
    cout << num.top() << endl;//最后结果

    return 0;
}

总结：该题可以联想到离散数学二叉树里的运算符中序遍历(用栈实现)，是一个模板题；可向unordered_map里添加其他运算符来实现衍生。
另外这题好像可以白嫖y总视频讲解哦~