题目链接:https://www.acwing.com/problem/content/3305/
题目
给定一个表达式,其中运算符仅包含 +,-,*,/
(加 减 乘 整除),可能包含括号,请你求出表达式的最终值。
注意
- 数据保证给定的表达式合法。
- 题目保证符号
-
只作为减号出现,不会作为负号出现,例如,-1+2
,(2+2)*(-(1+1)+2)
之类表达式均不会出现。
题目保证表达式中所有数字均为正整数。 - 题目保证表达式中所有数字均为正整数。
- 题目保证表达式在中间计算过程以及结果中,均不超过 2 31 − 1 2^{31}−1 231−1。
- 题目中的整除是指向 0 0 0 取整,也就是说对于大于 0 的结果向下取整,例如 5 / 3 = 1 5/3=1 5/3=1,对于小于 0 0 0 的结果向上取整,例如 5 / ( 1 − 4 ) = − 1 5/(1−4)=−1 5/(1−4)=−1。
- C++和Java中的整除默认是向零取整;Python中的整除
//
默认向下取整,因此Python的eval()
函数中的整除也是向下取整,在本题中不能直接使用。
注意:
输入格式
共一行,为给定表达式。
输出格式
共一行,为表达式的结果。
数据范围
表达式的长度不超过
1
0
5
10^5
105。
输入样例:
(2+2)*(1+1)
输出样例:
8
思路:
前缀表达式、中缀表达式、后缀表达式相当于一颗二叉树的先序遍历、中序遍历、后序遍历
其中内部节点(度为2)均为符号,叶子节点(度为0)均为数字
中序遍历,每次都要把子树算出来然后从下往上算
如何判断某棵子树被遍历完?
只要往上走就说明遍历完了,往上走等价于
当
前
运
算
符
的
优
先
级
<
=
上
一
个
运
算
符
的
优
先
级
当前运算符的优先级<=上一个运算符的优先级
当前运算符的优先级<=上一个运算符的优先级
只要某棵子树被遍历完,就可以把值算出来
遇到括号时特殊对待,遇到左括号直接入栈,遇到右括号就一直算直到左括号
AC代码
#include <iostream>
#include <cstring>
#include <algorithm>
#include <stack>
#include <unordered_map>
using namespace std;
stack<int> num;
stack<char> op;
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);
}
int main()
{
unordered_map<char,int> pr{{'+',1}, {'-',1}, {'*',2}, {'/',2}};
string str;
cin >> str;
for(int i = 0; i < str.size(); i++)
{
char c = str[i];
if(isdigit(c))
{
int x = 0,j = i;
while(j < str.size() && isdigit(str[j]))
x = x * 10 + str[j++] - '0';
i = j - 1;
num.push(x);
}
//括号特殊,遇到左括号直接入栈,遇到右括号计算括号里面的
else if(c == '(') op.push('(');
else if(c == ')')
{
while(op.top() != '(') eval();
op.pop();
}
else
{
while(op.size() && pr[op.top()] >= pr[c]) eval(); //待入栈运算符优先级低,则先计算
op.push(c);
}
}
while(op.size()) eval(); //剩余的进行计算
cout << num.top() << endl;
return 0;
}