Codeup Problem A: 简单计算器

原题地址

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;
}