[백준/c++] 28278번: 스택2

문제

정수를 저장하는 스택을 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램을 작성하시오.

명령은 총 다섯 가지이다.

1. 1 X: 정수 X를 스택에 넣는다. (1 ≤ X ≤ 100,000)
2. 2: 스택에 정수가 있다면 맨 위의 정수를 빼고 출력한다. 없다면 -1을 대신 출력한다.
3. 3: 스택에 들어있는 정수의 개수를 출력한다.
4. 4: 스택이 비어있으면 1, 아니면 0을 출력한다.
5. 5: 스택에 정수가 있다면 맨 위의 정수를 출력한다. 없다면 -1을 대신 출력한다.

입력

첫째 줄에 명령의 수 N이 주어진다. (1 ≤ N ≤ 1,000,000)

둘째 줄부터 N개 줄에 명령이 하나씩 주어진다.

출력을 요구하는 명령은 하나 이상 주어진다.

출력

출력을 요구하는 명령이 주어질 때마다 명령의 결과를 한 줄에 하나씩 출력한다.

예제 입력 1 

9
4
1 3
1 5
3
2
5
2
2
5

예제 출력 1 

1
2
5
3
3
-1
-1

정답

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

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    
    int N, cmd, num;
    stack<int> s;
    cin >> N;
    
    for (int i = 0; i < N; i++) {
        cin >> cmd;
        switch (cmd) {
            case 1:
                cin >> num;
                s.push(num);
                break;
            case 2:
                if (!s.empty()) {
                    cout << s.top() << '\n';
                    s.pop();
                } else {
                    cout << -1 << '\n';
                }
                break;
            case 3:
                cout << s.size() << '\n';
                break;
            case 4:
                cout << (s.empty() ? 1 : 0) << '\n';
                break;
            case 5:
                cout << (s.empty() ? -1 : s.top()) << '\n';
                break;
        }
    }
    
    return 0;
}

먼저, `stack<int> s;`를 통해 정수를 담을 스택을 선언하고 명령의 수를 입력받는다. 명령의 수만큼 반복문을 실행하면서, 각각의 명령을 처리합니다. case 1은 정수를 스택에 넣는 작업으로 이 때 추가할 숫자를 입력받아 `s.push(num);`을 통해 스택에 넣다. case 2는 스택에서 정수를 빼는 작업으로 스택이 비어있지 않다면, `s.top()`을 통해 스택의 맨 위에 있는 숫자를 출력하고, `s.pop();`을 통해 그 숫자를 스택에서 제거한다. 스택이 비어있다면 -1을 출력한다. case 3은 스택에 들어있는 정수의 개수를 출력하는 작업으로 `s.size()`를 통해 스택의 크기를 출력한다. case 4는 스택이 비어있는지 확인하는 작업으로 `s.empty()`를 통해 스택이 비어있는지 확인하고, 비어있다면 1을, 아니라면 0을 출력한다. case 5는 스택에 정수가 있다면 맨 위의 정수를 출력하는 작업으로 스택이 비어있지 않다면, `s.top()`을 통해 스택의 맨 위에 있는 숫자를 출력한다. 스택이 비어있다면 -1을 출력한다. 

 

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