백준 1935 - 후위표기식2

문제

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

1935번: 후위표기식2

 

풀이

후위표기식으로 된 식을 계산하는 문제이다.

1. 피연산자를 만나면 스택에 push한다.

2. 연산자를 만나면 스택에 있는 두 개의 피연산자를 꺼내서 연산을 한다.

3. 연산 값을 다시 스택에 push한다.

 

빼기(-) 와 나누기(/)를 할 때 피연산자의 순서에 주의한다.

 

cpp에서 소수점 2자리까지 표현하는 방법은 다음과 같다.

cout << fixed; : 소수점 아래 값을 고정하는 표현
cout.precision(2); : 소수점 아래를 2로 고정

 

코드

#include <iostream>
#include <stack>
#include <string>
 
using namespace std;
 
int N;
string post;
int number[26];
stack<double> s;
double result = 0;
 
int main()
{
    cin.tie(0);
    ios::sync_with_stdio(false);
 
    cin >> N;
    cin >> post;
 
    for (int i = 0; i < N; i++) cin >> number[i];
 
    for (int i = 0; i < post.size(); i++)
    {
        if (post[i] >= 'A' && post[i] <= 'Z') s.push(number[post[i] - 'A']);
        else
        {
            double op1 = s.top(); s.pop();
            double op2 = s.top(); s.pop();
            switch (post[i])
            {
            case '+':
                result = op1 + op2;
                break;
            case '-':
                result = op2 - op1;
                break;
            case '*':
                result = op1 * op2;
                break;
            case '/':
                result = op2 / op1;
                break;
            }
 
            s.push(result);
        }
    }
 
    cout << fixed;
    cout.precision(2);
    cout << s.top();
    
    return 0;
}