queue&stack實現簡單的計算器
阿新 • • 發佈:2018-12-30
晚上做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;
}