[정올] 1102 / 스택 (stack)

문제

i가 들어오면 숫자를 받아 stack에 push하고

o가 들어온다면 stack의 가장 위에있는 원소를 출력한뒤 재거

c가 들어온다면 stack의 크기를 출력하는 프로그램을 작성하면 된다.

해결 방법

std::stack을 알면 쉽게 해결 할 수 있다.

정답 코드

#include <bits/stdc++.h>
using namespace std;
int main(){
    int n;
    stack<int> st;
    cin >> n;
    while(n--){
        char x;
        cin >> x;
        if(x == 'i'){
            int g;
            cin >> g;
            st.push(g);
        }
        if(x == 'c'){
            cout << st.size() << "\n";
        }
        if(x == 'o'){
            if(st.empty()) cout << "empty\n";
            else {
                cout << st.top() << "\n";
                st.pop();
            }
        }
    }
}