一、学习代码(C++)
#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() && op.top() != '(' && pr[op.top] >= pr[c]) eval();
op.push(c);
}
}
while (op.size()) eval();
cout << num.top() << end1;
return 0;
}
手动运行
二、总结
该代码巧妙地结合了两个栈储存不一样的变量,再引用eval函数将两个栈联系起来。其中for函数嵌套if函数,if函数再嵌套while函数需要自己画图理解。
在此非常佩服学长的代码。