AcWing 3302. 表达式求值
原题链接
简单
作者:
小笨比
,
2021-03-14 02:20:43
,
所有人可见
,
阅读 593
#pragma GCC optimize(2)
#include <iostream>
#include <unordered_map>
#include <stack>
using namespace std;
void eval(stack<int> &num, stack<char> &op)
{
// 从操作数栈顶取两个数,操作符栈顶取一个操作符,进行运算
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);//将运算结果压入操作数栈中
}
int main()
{
cin.tie(0);
ios::sync_with_stdio(false);
// 设置加减乘除的优先级
unordered_map<char, int> pr{{'+', 1}, {'-', 1}, {'*', 2}, {'/', 2}};
stack<char> op;
stack<int> num;
string s;
cin >> s;
for(int i = 0; i < s.size(); i ++ )
{
char c = s[i];
if(isdigit(c))
{
int j = i, x = 0;// 先用j保存i,然后向后检查是否还有数字,是就将所有数字合并成一个数
while(j < s.size() && isdigit(s[j]))
x = x * 10 + s[j ++ ] - '0';
num.push(x);
i = j - 1;// while中j最后会自增一次,需要减1,满足for中i的自增
}
else if(c == '(') op.push(c);
else if(c == ')')
{
// 遇到右括号,就一直进行运算,直到遇到左括号
while(op.top() != '(') eval(num, op);// 有右括号就一定有左括号,所以不用判空,最后至少会有一个左括号在栈中
op.pop();// 弹出左括号
}
else
{
// 如果栈顶运算符优先级高与待入栈的运算符,则将栈顶运算符弹出,进行运算
while(!op.empty() && pr[op.top()] >= pr[c]) eval(num, op);// 如果op.top()是'(',那么pr['(']=0
op.push(c);
}
}
// 最后操作符栈中不含括号,但可能还有运算符,继续弹栈运算
while(!op.empty()) eval(num, op);
cout << num.top() << endl;
return 0;
}