백준 10828 c++ 스택

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

10828번: 스택

 

 

#include <iostream>
#include <string>
#include <stack>
using namespace std;
 
 
int main()
{
    int count;
    cin >> count; //몇 회 반복할지 입력 받는다
    stack<int> A; //스택 하나 선언
    string S; //명령어 입력받을 string 
    int Sn; //push 입력받았을 때 입력할 숫자
    while (count--) //입력받은 총 count 반복
    {
        cin >> S;
        if (S == "push") //입력받은 string이 push 일 경우 추가로 숫자 입력
        {
            cin >> Sn;
            A.push(Sn);
        }
        else if (S == "top") //스택이 비었을 경우 -1 출력
        {
            if (A.empty() == 1)
            {
                cout << -1 << endl;
            }
            else //안비었을 경우 스택의 top 출력
            {
                cout << A.top() << endl;
            }
        }
        else if (S == "size")//스택의 size 출력
        {
            cout << A.size() << endl;
        }
        else if (S == "empty")//스택이 비었을 경우 1출력 
        {
            if (A.empty() == 1)
            {
                cout << 1 << endl;
            }
            else//스택이 안 비었을 경우 0출력
            {
                cout << 0 << endl;
            }
        }
        else if (S == "pop")
        {
            if (A.empty() == 1) //스택이 비었을 경우 -1 출력
            {
                cout << -1 << endl;
            }
            else//아닐경우 top 출력하고 pop
            {
                cout << A.top() << endl;
                A.pop();
            }
        }
    }
}