给定一个表达式,其中运算符仅包含 +,-,*,/
(加 减 乘 整除),可能包含括号,请你求出表达式的最终值。后续可扩展运算符
/*
栈:表达式求值
*/
#include <iostream>
#include <cstring>
#include <algorithm>
#include <stack>
#include <unordered_map>
using namespace std;
stack<int> num; //数
stack<char> op; //运算符
// 运算操作,可扩展其他运算,eg:^
// 计算上一运算符
void eval()
{
// 注意a,b顺序 如果是a-b,是先取b再取a
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 if(c == '/') 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[op.top()] >= pr[c]) eval(); // 1*1+1
op.push(c); // 1+1*1
}
}
while(op.size()) eval();
cout << num.top() << endl;
return 0;
}