[Silver 4] 10828번 스택

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

10828번: 스택

 

이 문제는 STL의 vector를 이용해서 스택을 만들면 해결할 수 있다.

string 객체를 생성해 입력을 받아 if 문을 통해 명령이 무엇인지 판단한다.

그 후 명령에 따라 vector의 함수를 호출하면 된다.

push 명령의 경우 push_back을 사용한다.

pop의 경우 비어있는지 알 수 있는 empty를 사용하여 스택이 비어있는지 확인한 후 요소가 있다면 가장 최근에 삽입된 요소를 알 수 있는 back 함수와 꺼내는 pop_back을 사용한다.

size의 경우 size함수를 사용하고, empty의 경우 empty 함수를 사용한다.

마지막으로 top의 경우 pop 명령에서 pop_back 함수를 제외하고 똑같이 작성하면 된다.

 

#include<iostream>
#include<vector>
using namespace std;

int main(){
    ios::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
    int N;
    cin>>N;
    vector<int> stack;
    for(int i = 0;i<N;i++){
        string command;
        cin>>command;
        if(command =="push"){
            int input;
            cin>>input;
            stack.push_back(input);
        }
        if(command == "pop"){
            if(stack.empty()) cout<<-1<<'\n';
            else{
                int value = stack.back();
                stack.pop_back();
                cout<<value<<'\n';
            }
        }
        if(command == "size"){
            cout<<stack.size()<<'\n';
        }
        if(command == "empty"){
            cout<<stack.empty()<<'\n';
        }
        if(command == "top"){
            if(stack.empty()) cout<<-1<<'\n';
            else{
                int value = stack.back();
                cout<<value<<'\n';
            }
        }
    }
}