3302.表达式求值 做题笔记

两个核心关键点：

（1）双栈，一个操作数栈，一个运算符栈；

（2）运算符优先级，栈顶运算符和即将入栈的运算符的优先级比较：

如果栈顶的运算符优先级低，新运算符直接入栈；

如果栈顶的运算符优先级高，先出栈计算，新运算符再入栈。

以1+2+3*4*5举例，看是如何利用上述两个关键点实施计算的。

首先，这个例子只有+和*两个运算符，所以它的运算符表是：

这里的含义是：

（1）如果栈顶是+，即将入栈的是+，栈顶优先级高，需要先计算，再入栈；

（2）如果栈顶是+，即将入栈的是*，栈顶优先级低，直接入栈；

（3）如果栈顶是*，即将入栈的是+，栈顶优先级高，需要先计算，再入栈；

（4）如果栈顶是*，即将入栈的是*，栈顶优先级高，需要先计算，再入栈；

一开始，初始化好输入的字符串，以及操作数栈，运算符栈。

一步步，扫描字符串，操作数一个个入栈，运算符也入栈。

下一个操作符要入栈时，需要先比较优先级。

栈内的优先级高，必须先计算，才能入栈。

计算的过程为：

（1）操作数出栈，作为num2；

（2）操作数出栈，作为num1；

（3）运算符出栈，作为op；

（4）计算出结果；

结果入操作数栈；

接下来，运算符和操作数才能继续入栈。下一个操作符要入栈时，继续比较与栈顶的优先级。

栈内的优先级低，可以直接入栈。

字符串继续移动。

又要比较优先级了。

栈内的优先级高，还是先计算（3*4=12），再入栈。

不断入栈，直到字符串扫描完毕。

不断出栈，直到得到最终结果3+60=63，算法完成。

这个方法的时间复杂度为O(n)，整个字符串只需要扫描一遍。

代码：

#include<iostream>
#include<unordered_map>
#include<stack>
#include<string>
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}};//各个运算符优先级储存在哈希表里
    string str;
    cin>>str;
    for(int i=0;i<str.size();i++){//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;//i从这个数字下一位开始继续遍历
            num.push(x);//这个数字入数字栈
        }
        else if(c=='(') op.push(c);//左括号入符号栈
        else if(c==')'){//如果是右括号
            while(op.top()!='(') eval();//一直计算到遇到左括号为止
            op.pop();//左括号出栈
        }
        else {//如果c是其他运算符
            while(op.size()&&pr[op.top()]>=pr[c]) eval();//与栈顶运算符进行优先级比较，栈顶优先级高就先计算，再入栈
            op.push(c);
        }
    }
    while(op.size()) eval();//如果还有运算符，继续计算
    cout<<num.top()<<endl;//输出结果
}