[BOJ/Silver 4] 백준 28278 스택 2(C++)

문제 링크

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

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

 

 

알고리즘 분류

• 자료 구조

 

풀이

먼저 스택을 하나 선언하고, 명령에 따라 작업을 수행한다.

• 1번 명령이 주어지면 X를 추가로 입력받고 스택에 push한다.
• 2번 명령이 주어지면 스택의 top을 출력하고 pop한다. 스택이 비어있다면 -1을 출력한다.
• 3번 명령이 주어지면 스택에 있는 원소의 개수를 출력한다.
• 4번 명령이 주어지면 스택이 비어있다면 1을 출력하고 아니라면 0을 출력한다.
• 5번 명령이 주어지면 스택의 top을 출력한다. 스택이 비어있다면 -1을 출력한다.

 

코드

#include <iostream>
#include <vector>
#include <string>
#include <stack>
#include <algorithm>
#define FASTIO cin.tie(0); cout.tie(0); ios::sync_with_stdio(false);

using namespace std;
int N, C, X;
stack<int> S;

void input() {
    cin >> N;
    while (N--) {
        cin >> C;
        if (C == 1) {
            cin >> X;
            S.push(X);
        }
        else if (C == 2) {
            if (S.empty()) {
                cout << "-1\n";
            }
            else {
                cout << S.top() << "\n";
                S.pop();
            }
        }
        else if (C == 3) {
            cout << (int)S.size() << "\n";
        }
        else if (C == 4) {
            if (S.empty()) {
                cout << "1\n";
            }
            else {
                cout << "0\n";
            }
        }
        else if (C == 5) {
            if (S.empty()) {
                cout << "-1\n";
            }
            else {
                cout << S.top() << "\n";
            }
        }
    };
}

int main() {
    FASTIO
    input();

    return 0;
}