PTA 简单计算器

题目

本题要求你为初学数据结构的小伙伴设计一款简单的利用堆栈执行的计算器。如上图所示，计算器由两个堆栈组成，一个堆栈 $S_1$​ 存放数字，另一个堆栈 $S_2$​ 存放运算符。计算器的最下方有一个等号键，每次按下这个键，计算器就执行以下操作：

1. 从 $S_1$​ 中弹出两个数字，顺序为 $n_1$​ 和 $n_2$​；
2. 从 $S_2$​ 中弹出一个运算符 op；
3. 执行计算 $n_2$​ op $n_1$​；
4. 将得到的结果压回 $S_1$​。

直到两个堆栈都为空时，计算结束，最后的结果将显示在屏幕上。

输入

输入首先在第一行给出正整数 N（$1<N\le 10^3$），为 $S_1$​ 中数字的个数。
第二行给出 N 个绝对值不超过 100 的整数；第三行给出 N−1 个运算符 —— 这里仅考虑 +、-、*、/ 这四种运算。一行中的数字和符号都以空格分隔。

输出

将输入的数字和运算符按给定顺序分别压入堆栈 $S_1$​ 和 $S_2$​，将执行计算的最后结果输出。注意所有的计算都只取结果的整数部分。题目保证计算的中间和最后结果的绝对值都不超过 $10^9$。
如果执行除法时出现分母为零的非法操作，则在一行中输出：$ERROR:X/0$，其中 X 是当时的分子。然后结束程序。

输入样例1

5
40 5 8 3 2
/ * - +

输出样例1

2

输入样例2

5
2 5 8 4 4
* / - + 

输出样例2

ERROR: 5/0

题目分析

本题分析后，发现可以利用两个栈来分别存储操作数和操作，并且在操作数栈中，先弹出的作为$S_2$，计算出结果后，再压入栈中，做为下次运算的操作数。当操作栈为空时，操作数栈中也仅剩一位数据，此时便是答案，途中遇到除以零的情况则提前结束。

代码

#include<iostream>
#include<stack>
using namespace std;
int main(){
	stack<int> nu;//number
	stack<char> op;//operator
	int n,k;
	char p;
	cin >> n;
	for(int i = 0;i<n;++i){
		cin >> k;
		nu.push(k);
	}
	for(int i = 0;i<n-1;++i){
		cin >> p;
		op.push(p);
	}
	int ans,num1,num2,ope;
	while(!op.empty()){
		num2=nu.top();
		nu.pop();//读取操作数
		num1=nu.top();
		nu.pop();
		ope = op.top();
		op.pop();
		if(ope=='+'){
			ans = num1+num2;
			nu.push(ans);
		}
		else if(ope=='-'){
			ans = num1-num2;
			nu.push(ans);
		}
		else if(ope=='*'){
			ans = num1*num2;
			nu.push(ans);
		}
		else if(ope == '/'){
			if(num2==0){//判零
				cout <<"ERROR: "<<num1<<"/0"<<endl;
				return 0;
			}
			ans=num1/num2;
			nu.push(ans);
		}
	}
	ans = nu.top();
	cout << ans << endl;
	return 0;
}

运行结果