[백준] 1935 후위 표기식 2

[백준] 1935 후위 표기식 2

• https://www.acmicpc.net/problem/1935

#include <algorithm>
#include <string>
#include <stack>
#include <vector>
#include <iostream>
using namespace std;

int main() {
	ios_base::sync_with_stdio(false);
	cin.tie(NULL); cout.tie(NULL);

	int n;
	cin >> n;

	string str;
	cin >> str;

	vector<double> alpha;
	for (int i = 0; i < n; ++i) {
		double input;
		cin >> input;
		alpha.push_back(input);
	}

	stack<double> st;
	for (int i = 0; i < str.length(); ++i) {
		char ch = str[i];
		//연산자인 경우
		if (ch == '+') {
			double db2 = st.top();
			st.pop();
			double db1 = st.top();
			st.pop();

			st.push(db1 + db2);

		}
		else if (ch == '-') {
			double db2 = st.top();
			st.pop();
			double db1 = st.top();
			st.pop();

			st.push(db1 - db2);
		}
		else if (ch == '*') {
			double db2 = st.top();
			st.pop();
			double db1 = st.top();
			st.pop();

			st.push(db1 * db2);
		}
		else if (ch == '/') {
			double db2 = st.top();
			st.pop();
			double db1 = st.top();
			st.pop();
			st.push(db1 / db2);
		}
		//피연산자인 경우
		else {
			//알파벳 -> 대응하는 숫자로 치환하여 스택에 저장
			st.push(alpha[ch - 'A']);
		}
	}

	//소수점 아래 2자리까지 출력
	cout << fixed;
	cout.precision(2);
	cout << st.top();

	return 0;
}