[BOJ] 후위 표기식2 1935번

문제

후위 표기식과 각 피연산자에 대응하는 값들이 주어져 있을 때, 그 식을 계산하는 프로그램을 작성하시오.

입력

첫째 줄에 피연산자의 개수(1 ≤ N ≤ 26) 가 주어진다. 그리고 둘째 줄에는 후위 표기식이 주어진다. (여기서 피연산자는 A~Z의 영대문자이며, A부터 순서대로 N개의 영대문자만이 사용되며, 길이는 100을 넘지 않는다) 그리고 셋째 줄부터 N+2번째 줄까지는 각 피연산자에 대응하는 값이 주어진다. 3번째 줄에는 A에 해당하는 값, 4번째 줄에는 B에 해당하는값 , 5번째 줄에는 C ...이 주어진다, 그리고 피연산자에 대응 하는 값은 100보다 작거나 같은 자연수이다.

후위 표기식을 앞에서부터 계산했을 때, 식의 결과와 중간 결과가 -20억보다 크거나 같고, 20억보다 작거나 같은 입력만 주어진다.

출력

계산 결과를 소숫점 둘째 자리까지 출력한다.

 

접근 방식

 

오랜만에 재밌게 푼 문제였다!

stack을 이용하여 간단하게 풀 수 있다

후위 표기식에서 반복문으로 각 문자를 확인하면서 문자가 만약 알파벳이라면 stack에 넣어주고

연산자라면  stack에서 두개의 숫자를 꺼내 연산자에 맞는 연산을 한 후 stack에 넣어주면 된다!

 

그런데 주의할 점은 소수점 둘째 자리까지 출력해야 하므로

C++에서는 cout<<fixed<<setprecision(2) 을 사용해 출력할 수 있다!

 

코드

#include<bits/stdc++.h>
using namespace std;
typedef long double ld;

int main() {
    ios::sync_with_stdio(false); cin.tie(0);
    int alpha[26];
    int N; cin>>N;
    string exp; cin>>exp;
    vector<char> v;
    
    for(int i=0; i<N; i++) {
        int x; cin>>x;
        alpha[i] = x;
        
    }
    ld ans = 0;

    stack<ld> st;

    for(auto c : exp) {
        if(c == '*') {
            auto x = st.top();
            st.pop();
            auto y = st.top();
            st.pop();
            ans = y * x;
            st.push(ans);
        }
        else if(c == '-') {
            auto x = st.top();
            st.pop();
            auto y = st.top();
            st.pop();
            ans = y - x;
            st.push(ans);

        }
        else if(c == '/') {
            auto x = st.top();
            st.pop();
            auto y = st.top();
            st.pop();
            ans = y / x;
            st.push(ans);
        }

        else if(c == '+') {
            auto x = st.top();
            st.pop();
            auto y = st.top();
            st.pop();
            ans = y + x;
            st.push(ans);
        }

        else {
            st.push(alpha[c - 'A']);
        }
    }


    cout.precision(2);
    cout<<fixed<<setprecision(2)<<st.top();


}