原题链接如下:
AcWing 3302. 表达式求值
其实这也就是大家耳熟能详的用栈实现计算器,也许很多人包括我自己,都曾经尝试过去实现,最后即使成功了也感觉代码质量不行,思路不够清晰,改bug改到吐血
这里是y总大佬实现中缀表达式计算(计算器)的清晰思路,代码非常优雅清晰
同时对中缀表达式、用栈实现计算器等相关事项和原理都不太了解的,也可以看这位哥们的讲解,简直保姆级
我自己的代码里也有对每个细节的详细注释
#include<iostream>
#include<string>
#include<unordered_map>
#include<algorithm>
#include<stack>
using namespace std;
//定义数字栈和操作符栈
stack<int>num;
stack<char>op;
//定义优先级表
unordered_map<char,int> h {{'+',1},{'-',1},{'*',2},{'/',2}};
//定义一下进行运算操作的函数eval
void eval()
{
//开始把数字栈中弹出两个数字
//请注意,由于是栈存储,所以我第一个取出来是第二个数字
//后者数字
int b = num.top();
num.pop();
//前者数字
int a = num.top();
num.pop();
//取出操作符
char p = op.top();
op.pop();
//定义结果
int r = 0;
//计算结果
if(p == '+') r = a+b;
if(p == '-') r = a-b;
if(p == '*') r = a*b;
if(p == '/') r = a/b;
//将运算结果压入栈中
num.push(r);
}
int main()
{
string s;//读入表达式
cin>>s;
for(int i = 0;i<s.size();i++)//开始分解表达式
{
if(isdigit(s[i]))//数字入栈
{
int x = 0,j = i;
while(j<s.size()&&isdigit(s[j]))
{
x = x*10 + s[j] - '0';
j++;
}
num.push(x);
i = j-1;//由于while最后那里有一个j++ 多加了一次 i就需要减回去
}
//如果遇到左括号
else if(s[i] == '(')//左括号入栈
{
op.push(s[i]);
}
//如果遇到右括号
else if(s[i] == ')')
{
while(op.top()!='(')//直到遇到左括号
eval();//把数字栈顶两个数和栈顶符号一起运算一次
op.pop();//左括号出栈
}
//如果遇到操作符
//操作符要么就比栈顶优先级低 则栈顶运算一波
//操作符要么比栈顶优先级高 就直接进
else
{
while(op.size()&&h[op.top()]>=h[s[i]])//属于栈顶优先级更高,运算一波
{
eval();
}
//此时就是栈顶优先级更低,操作符可以直接入栈
op.push(s[i]);
}
}
//此时表达式已经全部分解完毕,但是可能这两个栈里面还有残留的运算符和操作数
while(op.size())eval();//剩余的进行计算
//此时数字栈的栈顶就是答案
cout<<num.top()<<endl;
return 0;
}