codeup墓地 问题 A: 简单计算器

codeup墓地 问题 A: 简单计算器

// An highlighted block
#include<iostream>
#include<queue>
#include<stack>
#include<map>
#include<string>
#include<cstring>
#include<iomanip>
using namespace std;
//p用来设置操作符的优先级,对应的键值对的建立在main里面完成 
map<char, int> p;
//中缀表达式变成后缀表达式 
void dealpre(string &pre, string &bac, stack<char> &ope){
	int i, prelen = pre.size();
	for(i = 0; i < prelen; i++){
		if(('0' <= pre[i] && pre[i] <= '9') || pre[i] == ' '){
			bac += pre[i]; 
		} else if(pre[i] == '+' || pre[i] == '-' || pre[i] == '*' || pre[i] == '/'){
			if(ope.empty() || p[pre[i]] > p[ope.top()]) ope.push(pre[i]);
			else{
				while(!ope.empty() && p[ope.top()] >= p[pre[i]]){
					bac += ope.top();
					ope.pop();
				}
				ope.push(pre[i]);
			}
		}
	}
	while(!ope.empty()){
		bac += ope.top();
		ope.pop();
	}
	return;
}
double calculate(string &bac, stack<double> &ans){
	int i, baclen = bac.size(); 
	double num = 0, temp1, temp2;
	bool flag = true;//flag为true代表是第一此出现空格
	bool judge = false; //judge为false代表最后一个数字还没入栈 
	for(i = 0; i < baclen; i++){
		if('0' <= bac[i] && bac[i] <= '9'){
			num = num * 10 + bac[i] - '0';
			flag = true;
		} else if(bac[i] == ' '){
			if(flag){ //这个flag要是和上面那个bac[i] == ' '写在一起
			//并用 && 连接, 会导致下面那个else代表的不是 (bac[i] < '0' || bac[i] > '9')
			// && bac[i] != ' ' 而是 (bac[i] < '0' || bac[i] > '9') && (bac[i] != ' ' || flag == false)
			//导致不符合原本用意 
				ans.push(num);
				num = 0;
				flag = false;
			}
		} else{
			if(!judge && (i == baclen - 2 || i == baclen - 1)){
			//后缀表达式最后一个数之后只有运算符,没有空格,而且可能是有一个运算符或者有两个运算符,而这个数只需入栈一次,故这个if下的代码只能做一次,因此借助judge的值来判断是否已经做过
				ans.push(num);
				num = 0;
				judge = true;
			}
			if(!ans.empty()){
				temp2 = ans.top();
				ans.pop();
			} 
			if(!ans.empty()){
				temp1 = ans.top();
				ans.pop();	
			} 
			switch(bac[i]){
				case '+':
					ans.push(temp1 + temp2);
					break;
				case '-':
					ans.push(temp1 - temp2);
					break;
				case '*':
					ans.push(temp1 * temp2);
					break;
				case '/':
					ans.push(temp1 / temp2);
					break;
			}
		}
	}
	return ans.top();	
}
int main(){
	p['*'] = 1;
	p['/'] = 1;
	p['+'] = 0;
	p['-'] = 0;
	string pre;
	getline(cin, pre);
	while(pre != "0"){
		string bac;
		stack<char> ope;
		stack<double> ans;
		//pre每次输入得来,会刷新;bac,ope,ans定义在这个循环里面可以免除手动重置步骤 
		dealpre(pre, bac, ope);
		cout << setiosflags(ios::fixed) << setprecision(2) << calculate(bac, ans) << endl;
		getline(cin, pre);
	}
	return 0;
}