栈 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;
}
原创不易
转载请标明出处
如果对你有所帮助 别忘啦点赞支持哈