原题地址
http://codeup.hustoj.com/problem.php?cid=100000605&pid=0
解题思路
这题搞好久…
1.读取数字和操作符,我是把整个式子当成一个字符串读入,然后根据空格的位置截取子串,判断其是操作符还是操作数。
2.三个东西要弄清楚。
1)一个是操作符栈op
,用来暂时存放读进来的操作符,之后读进来的如果是操作符,就和这个栈顶的比较一下优先级,如果栈顶的优先级更高或者与之相等就弹出。
2)一个是队列q
,用来存储我们从原来的中缀表达式转化过来的后缀表达式。这个后缀表达式,是在我们读入的时候弄的,过程如下:
遇到一个操作数,直接加入队列;遇到一个操作符,经过上面op
相关的操作,可能会有栈顶的元素加入队列。
3)一个是操作数栈num
,用来存放最后处理q
中的后缀表达式时遇到的操作数。过程如下:
最后东西都读完了也存完拉,我们就开始计算这个后缀表达式。这个是一个队列,就从头开始一直弹出来即可。
如果得到的是操作数,就先压入num
;如果是操作符,则进行计算,弹出最前面的两个数,计算结果再压回num
,以此类推。最终留在栈顶的元素即为答案~
注意事项
1.涉及到数字的直接用double
,省事。
2.中缀表达式处理完之后,要记得把操作符栈op里面的操作符都弹到q里去,不然就漏拉~
参考代码
#include <bits/stdc++.h>
using namespace std;
#define pb push_back;
typedef double db;
typedef long long LL;
typedef vector<int> VI;
const int inf = 2e9;
const LL INF = 8e18;
const int maxn = 5e5 + 5;
double cal(double a, double b, char c) {
if (c == '/') return a / b;
if (c == '-') return a - b;
if (c == '+') return a + b;
if (c == '*') return a * b;
}
bool judge(char c) {
if (c >= '0' && c <= '9') return false;
else return true;
}
struct node{
double num;
char op;
bool flag;
};
int main() {
stack<char> op;
map<char, int> pr{{'+', 0}, {'-', 0}, {'/', 1}, {'*', 1}};
queue<node> q;
string s;
while (1) {
getline(cin, s);
if (s.size() == 1 && s[0] == '0') break;
double pre = 0;
string item;
bool flag = true;
while (1) {
int i;
for (i = 0; i < s.size(); i++) {
if (s[i] == ' ') {
item = s.substr(0, i);
if (judge(item[0])) {
while (!op.empty() && pr[item[0]] <= pr[op.top()]) {
node now;
now.flag = false;
now.op = op.top();
q.push(now);
op.pop();
}
op.push(item[0]);
}
else {
node now;
now.flag = true;
now.num = stod(item);
q.push(now);
}
s.erase(0, i + 1);
break;
}
}
if (i == s.size()) {
node last;
last.flag = true;
last.num = stod(s);
q.push(last);
while (!op.empty()) {
node ne;
ne.flag = false;
ne.op = op.top();
q.push(ne);
op.pop();
}
break;
}
}
double temp1, temp2, tmp;
node cur;
stack<double> num;
while (!q.empty()) {
cur = q.front();
q.pop();
if (cur.flag) num.push(cur.num);
else {
temp2 = num.top();
num.pop();
temp1 = num.top();
num.pop();
tmp = cal(temp1, temp2, cur.op);
num.push(tmp);
}
}
printf("%.2f\n", num.top());
}
return 0;
}