题220.2022寒假天梯赛训练-7-9 简单计算器 (25 分)
题目:
本题要求你为初学数据结构的小伙伴设计一款简单的利用堆栈执行的计算器。如上图所示,计算器由两个堆栈组成,一个堆栈 S1存放数字,另一个堆栈S2,存放运算符。计算器的最下方有一个等号键,每次按下这个键,计算器就执行以下操作:
1.从S1中弹出两个数字,顺序为n1和n2;
2.从S2中弹出一个运算符op;
3.执行计算n2 op n1;
4.将得到的结果压回S1;
直到两个堆栈都为空时,计算结束,最后的结果将显示在屏幕上。
输入格式:
输入首先在第一行给出正整数 N(1<N≤10 3),为 S1中数字的个数。
第二行给出 N 个绝对值不超过 100 的整数;第三行给出 N−1 个运算符 —— 这里仅考虑 +、-、*、/ 这四种运算。一行中的数字和符号都以空格分隔。
输出格式:
将输入的数字和运算符按给定顺序分别压入堆栈 S1和S2 ,将执行计算的最后结果输出。注意所有的计算都只取结果的整数部分。题目保证计算的中间和最后结果的绝对值都不超过 10^9。
如果执行除法时出现分母为零的非法操作,则在一行中输出:ERROR: X/0,其中 X 是当时的分子。然后结束程序。
输入样例:
5
40 5 8 3 2
/ * - +
输出样例:
2
输入样例:
5
2 5 8 4 4
* / - +
输出样例:
ERROR: 5/0
思路:
照着题目说的,开两个stack,一个数栈,一个操作符栈,直接模拟就好了,最后的结果就是数栈栈顶的元素。代码如下:
代码:
#include <bits/stdc++.h>
using namespace std;
int main()
{
int N;
cin>>N;
stack<int> s_num;
for(int i=0;i<N;i++)
{
int num;
cin>>num;
s_num.push(num);
}
stack<char> s_op;
for(int i=0;i<N-1;i++)
{
char op;
cin>>op;
s_op.push(op);
}
while(!s_op.empty())
{
int n1=s_num.top();
s_num.pop();
int n2=s_num.top();
s_num.pop();
char op=s_op.top();
s_op.pop();
int res;
switch(op)
{
case '+':
res=n2+n1;
break;
case '-':
res=n2-n1;
break;
case '*':
res=n2*n1;
break;
case '/':
if(n1==0)
{
printf("ERROR: %d/%d",n2,n1);
return 0;
}
res=n2/n1;
break;
}
s_num.push(res);
}
printf("%d",s_num.top());
}