栈 AcWing 3302. 表达式求值

原题链接

AcWing 3302. 表达式求值

算法标签

栈 表达式求值

思路

摘自该题解

代码

#include<bits/stdc++.h>
#define int long long
#define rep(i, a, b) for(int i=a;i<b;++i)
#define Rep(i, a, b) for(int i=a;i>b;--i)
using namespace std;
const int N = 10005;
stack<int> num;
stack<int> op;
inline int read(){
   int s=0,w=1;
   char ch=getchar();
   while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
   while(ch>='0'&&ch<='9') s=s*10+ch-'0',ch=getchar();
   return s*w;
}
void put(int x) {
    if(x<0) putchar('-'),x=-x;
    if(x>=10) put(x/10);
    putchar(x%10^48);
}
void eval(){
    int b=num.top();
    num.pop();
    int a=num.top();
    num.pop();
    char 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);
}
signed main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    // 定义运算符优先级
    unordered_map<char, int> ump={
   {'+', 1}, {'-', 1}, {'*', 2}, {'/', 2}};
    string s;
    cin>>s;
    rep(i, 0, s.size()){
    	// 提取字符串中数字 
        if(isdigit(s[i])){
            int j=i, x=0;
            while(j<s.size()&&isdigit(s[j])){
                x=x*10+s[j++]-'0';
            }
            i=j-1;
            num.push(x);
        }else if(s[i]=='('){ // ( 入栈
            op.push(s[i]);
        }else if(s[i]==')'){ // ) 与 ( 之间数字进行运算
            while(op.top()!='('){
                eval();
            }
            // ) 出栈
            op.pop();
        }else{ // + - * / 运算  待入栈运算符优先级低，则先计算 后将计算结果入栈
            while(op.size()&&op.top()!='('&&ump[op.top()]>=ump[s[i]]){
                eval();
            }
            // 否则 先入栈 后计算
            op.push(s[i]);
        }
    }
    // 将所有未操作运算符 从前至后操作
    while(op.size()){
        eval();
    }
    printf("%lld", num.top());
    return 0;
}

原创不易
转载请标明出处
如果对你有所帮助 别忘啦点赞支持哈