题目来源:AcWing 3302. 表达式求值
一、题目描述
给定一个表达式,其中运算符仅包含 +,-,*,/
(加 减 乘 整除),可能包含括号,请你求出表达式的最终值。
注意:
- 数据保证给定的表达式合法。
- 题目保证符号
-
只作为减号出现,不会作为负号出现,例如,-1+2
,(2+2)*(-(1+1)+2)
之类表达式均不会出现。 - 题目保证表达式中所有数字均为正整数。
- 题目保证表达式在中间计算过程以及结果中,均不超过 2 31 − 1 2^{31}−1 231−1。
- 题目中的整除是指向 0 0 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
二、算法思路
中缀表达式可以转换为一棵表达式树,其中叶结点都是都是数字,其他的结点都是运算符。如果我们可以根据中缀表达式建立出一棵表达式树,那么按照中序遍历的递归顺序完全可以计算出来整个表达式的值。但是实际操作的时候我们不需要真的建立出这么一棵树,我们可以通过使用一些性质,比如说栈来达到同样的中缀表达式的计算效果。
中缀表达式各子树的计算顺序如上图所示。
在递归计算中缀表达式树的时候,我们计算一个结点的值时会先递归计算左子树的值,然后将该值和当前结点的运算符都记录下来(数值栈和符号栈)。继续递归计算右子树的值,当左右子树的值都计算出来以后,才执行左子树结果 op 右子树结果
的计算。
那么,如何判断某个子树已经计算完成?如果从表达式树的角度来看,计算方向往上走说明当前子树已经计算完;往下走说明某个子树还没有算完,非常容易判断。但是如何通过运算符栈来判断呢?我们可以通过表达式树发现,当左子树先计算完往上走,遇到其父结点运算符。这个过程体现了父结点的运算符优先级比原来左子树根的运算符优先级小,如下图。
因此,当符号栈进栈的时候,如果进栈符号优先级比栈顶符号小,说明左子树已经计算条件已经成熟,直接将栈顶符号弹出并计算,将计算结果压入数值栈,直到栈顶元素的优先级
<
<
<新进栈符号优先级,然后才让新来的符号入符号栈。此时数值栈的栈顶一定是新进栈符号的左子树的计算结果。如果进栈符号的优先级和符号栈栈顶符号相同,则按照“相同优先级从左往右算”的规则,此时同样进行上述操作,直到保证符号栈栈顶元素的优先级
<
<
<新进栈符号的优先级时才进栈。
因此,可以总结出:符号栈入栈符号优先级 ≤ ≤ ≤ 符号栈栈顶符号优先级,则不断让符号栈元素出栈,计算左子树值并压入数值栈,直到符号栈栈顶元素优先级 < < <新进栈元素优先级时,才让新符号入栈。整个符号栈必须时刻保持从栈底到栈顶符号优先级严格递增。
上面的规则只针对于没有括号的表达式。当存在括号时,如果遇到入栈符号是(
则直接入符号栈;如果入栈符号是)
则不断弹出符号栈,边弹出边计算,直至将第一个匹配的)
弹出为止,这样就能满足括号的优先性。
三、代码
#include <iostream>
#include <cstring>
#include <algorithm>
#include <stack>
#include <unordered_map>
using namespace std;
stack<int> num;
stack<char> op;
void eval()
{
auto b = num.top(); num.pop();
auto a = num.top(); num.pop();
auto 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++)
{
auto 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(c);
else if (c == ')')
{
while (op.top() != '(') eval();
op.pop();
}
else
{
while (op.size() && pr[c] <= pr[op.top()]) eval();
op.push(c);
}
}
while (op.size()) eval();
cout << num.top() << endl;
return 0;
}