[BOJ] 1935 후위표기식2 C++

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

1935번: 후위 표기식2

후위 표기식이 주어졌을 때 이를 계산하는 문제이다.

스택을 만들고 연산기호( * , / , + , - )가 오는 경우는 위에서 두개를 계산하고 push 그 외에는 그냥 push해주면 된다.

 

코드

#include <iostream>

#include <string>

#include <vector>

#include <stack>

#define ch 65

using namespace std;

string s;

double alpa[26];

 

int main(){

    int N; cin>>N;

    cin>>s;

    for(int i=0;i<N;i++) cin>>alpa[i];

    stack<double> st;

    for(int i=0;i<s.size();i++){

        if(s[i]=='+'||s[i]=='-'||s[i]=='/'||s[i]=='*'){

            double a2=st.top();

            st.pop();

            double a1=st.top();

            st.pop();

            if(s[i]=='+') st.push(a1+a2);

            else if(s[i]=='-') st.push(a1-a2);

            else if(s[i]=='*') st.push(a1*a2);

            else if(s[i]=='/') st.push(a1/a2);

        }else{

            st.push(alpa[int(s[i])-ch]);

        }

    }

    cout<<fixed;

    cout.precision(2);

    cout<<st.top();

    return 0;

}