1. 程式人生 > >queue&stack實現簡單的計算器

queue&stack實現簡單的計算器

晚上做PAT半個小時沒有把失去的3分找回來(17 / 20), 轉去刷codeup, 這個OJ和PAT有一些不同,它進行的是多點測試,難度要大於PAT,沒有煩人的PAT格式要求,可以把更多的精力放在演算法本身。

下面是[codeup 1918]簡單計算器的題目,對於四則運算,主要就是兩步,中綴表示式轉換為字尾表示式,字尾表示式求值。這屬於資料結構中stack和queue的應用。為了把更多的精力放在演算法實現,直接呼叫了C++ STL中的queue和stack。
如果對演算法原理不懂,百度一下原理滿天飛,耐心看完一篇部落格就懂了……這裡不再贅述。
#include<iostream>
#include<cstdio> #include<string> #include<stack> #include<queue> #include<map> using namespace std; //抽象運算元和操作符,代價是消耗記憶體 typedef struct Node { double num; //運算元 char op; //操作符 bool flag; //運算元true, 操作符false }Node; string str; //獲取計算表示式 stack<Node>
s; //操作符棧 queue<Node> q; //字尾表示式序列 map<char, int> op; void change() { double num; Node tmp; for(int i = 0; i < str.length(); ) { if(str[i] >= '0' && str[i] <= '9') { tmp.flag = true; tmp.num = str[i++] - '0'; while
(i < str.length() && str[i] <= '9' && str[i] >= '0') tmp.num = tmp.num * 10 + str[i++] - '0'; q.push(tmp); } else { tmp.flag = false; while(!s.empty() && op[s.top().op] >= op[str[i]]) { q.push(s.top()); s.pop(); } tmp.op = str[i]; s.push(tmp); ++i; } } while(!s.empty()) { q.push(s.top()); s.pop(); } } double calc() { Node tmp, cur; while(!q.empty()) { tmp = q.front(); q.pop(); if(tmp.flag) { s.push(tmp); } else { double num1, num2, res; num2 = s.top().num; s.pop(); num1 = s.top().num; s.pop(); switch(tmp.op) { case '+' : res = num1 + num2; break; case '-' : res = num1 - num2; break; case '*' : res = num1 * num2; break; case '/': res = num1 / num2; break; default: break; } cur.num = res; cur.flag = true; s.push(cur); } } return s.top().num; } int main() { op['+'] = op['-'] = 1; op['*'] = op['/'] = 2; while(getline(cin, str), str != "0") { //去除輸入的空格 for(string::iterator it = str.end(); it != str.begin(); --it) { if(*it == ' ') str.erase(it); } //初始化操作符棧 while(!s.empty()) s.pop(); //中綴表示式轉換為字尾表示式 change(); printf("%.2f", calc()); } return 0; }