中缀表达式求值——acWing

中缀表达式求值——acWing

题目背景

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

注意：

• 数据保证给定的表达式合法。
• 题目保证符号 - 只作为减号出现，不会作为负号出现，例如，-1+2,(2+2)*(-(1+1)+2) 之类表达式均不会出现。
• 题目保证表达式中所有数字均为正整数。
• 题目中的整除是指向 0 取整，也就是说对于大于 0 的结果向下取整，例如 5/3=1，对于小于 0 的结果向上取整，例如 5/(1−4)=−1。
• C++和Java中的整除默认是向零取整；Python中的整除//默认向下取整，因此Python的eval()函数中的整除也是向下取整，在本题中不能直接使用。

输入样例

(2+2)*(1+1)

输出样例

8

题目分析

由中缀表达式求值相比与由后缀表达式（逆波兰表达式）求值会更繁琐一点。

可由表达式树递归的求解表达式的值，也可以用栈模拟求值过程。

关键在于设定运算符的优先级（用unordered_map，注意不要漏掉‘（’），用来判断是否可以计算表达式中某个子树的结果。

代码如下

#include <bits/stdc++.h>

using namespace std;

const int N=1e5+10;

int stkNum[N],tt1=-1,tt2=-1;
char stkOp[N];

void eval(){
    int num2=stkNum[tt1--];
    int num1=stkNum[tt1--];
    char op=stkOp[tt2--];
    if(op=='+') stkNum[++tt1]=num1+num2;
    else if(op=='-') stkNum[++tt1]=num1-num2;
    else if(op=='*') stkNum[++tt1]=num1*num2;
    else if(op=='/') stkNum[++tt1]=num1/num2;
}

int main(){
    string str;
    ios::sync_with_stdio(false);
    cin.tie(0);
    cin>>str;
    unordered_map<char,int> pr={{'+',1},{'-',1},{'*',2},{'/',2}};
    for(int i=0;i<str.size();i++){
        char c=str[i];
        if(c=='(') stkOp[++tt2]='(';
        else if(c==')'){
            while(tt2!=-1&&stkOp[tt2]!='(')eval();
            tt2--;
        }
        else if(isdigit(c)){
            int x=0,j=i;
            while(j<str.size()&&isdigit(str[j])){
                x=10*x+str[j++]-'0';
            }
            stkNum[++tt1]=x;
            i=j-1;
        }
        else{ //若c为加减乘除
            while(tt2!=-1&&pr[stkOp[tt2]]>=pr[c]) eval();  //这句话为什么是while不是if？
            stkOp[++tt2]=c;
        }
    }
    
    while(tt2!=-1)eval();
    
    printf("%d",stkNum[tt1]);
    return 0;
}