Acwing_算法基础_数据结构_3302_表达式求值

AcWing 3302. 表达式求值

#include<iostream>
#include<stack>
#include<string>
#include<unordered_map>
using namespace std;
//定义优先级，与平常运算的优先级一样
/*
 * 考虑一个表达式 1+2+3*4*5
 * 两个栈 一个数字栈 一个符号栈
 * 如果符号栈顶的运算符优先级低，新运算符直接入栈
 * 如果符号栈顶的运算符优先级高，先出栈计算，新运算符再入栈
 * 相同符号先出栈计算，不同符号看优先级
 */
unordered_map<char,int> h={{'+',1},{'-',1},{'*',2},{'/',2}};
stack<int> nums;
stack<char> op;
//计算函数，得到的最终值要加入到数字栈中
void eval(){
    int a=nums.top();
    nums.pop();
    int b=nums.top();
    nums.pop();
    int p=op.top();
    op.pop();
    int r=0;
    if(p=='+') r=b+a;
    if(p=='-') r=b-a;
    if(p=='*') r=b*a;
    //唯一要想清楚的是谁除以谁
    if(p=='/') r=b/a;
    nums.push(r);
}
int main(){
    string s;
    cin>>s;
    for(int i=0;i<s[i];i++){
        if(isdigit(s[i])){
            //这是string类型转数值的方法
            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{
            if(op.size()&&h[op.top()]>=h[s[i]]) eval();
            op.push(s[i]);
        }
    }
    while(op.size()) eval();
    cout<<nums.top()<<endl;
    return 0;
}