【PTA】L2-033 简单计算器 (25 分)

L2-033
题目叙述和输入、输出格式等见上面的链接。

5
40 5 8 3 2
/ * - +

2

5
2 5 8 4 4
* / - +

ERROR: 5/0

由于顺序符号这些都已经是题目给定的，所以就不用进行其他处理，按照输入顺序入栈即可。

while循环结束的条件是操作符栈(optr)为空；若其不为空，则从操作数栈(opnd)中按顺序取出 n2 和 n1，运算结果重新入opnd栈（如果除数为0，则进行错误提示Error，然后直接return即可）。

循环结束后，操作数栈栈顶应该还剩下一个元素，输出它。

#include <iostream>
#include <stack>

using namespace std;

int main() {
	stack<int> opnd; // 操作数栈
	stack<char> optr; // 操作符栈
	
	int n;
	cin >> n;
	
	for (int i = 0; i < n; i++) {
		int n;
		cin >> n;
		opnd.push(n);
	}
	for (int i = 0; i < n - 1; i++) {
		char c;
		cin >> c;
		optr.push(c);
	}
	
	while (!optr.empty()) {
		int n1, n2;
		n1 = opnd.top(); opnd.pop();
		n2 = opnd.top(); opnd.pop();
		char c;
		c = optr.top(); optr.pop();
		int n3;
		switch (c) {
			case '+':
				n3 = n2 + n1;
				break;
			case '-':
				n3 = n2 - n1;
				break;
			case '*':
				n3 = n2 * n1;
				break;
			case '/':
				if (n1 == 0) {
					cout << "ERROR: " << n2 << "/0" << endl;
					return 0;
				}
				n3 = n2 / n1;
				break;
		}
		opnd.push(n3);
	}
	cout << opnd.top() << endl;
	
	return 0;
}