[백준 / C, C++] Day4. 스택 2 (28278)

2024. 05. 21 - 코딩테스트 스터디 Day4

 

 

01. 문제

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

명령은 총 다섯 가지이다.

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

 

 

02. 입력

첫째 줄에 명령의 수 N이 주어진다. (1 ≤ N ≤ 1,000,000)
둘째 줄부터 N개 줄에 명령이 하나씩 주어진다.
출력을 요구하는 명령은 하나 이상 주어진다.

 

 

03. 출력

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

 

 

04-1. 예제 입력 1

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

 

 

04-2. 예제 출력 1

1
2
5
3
3
-1
-1

 

 

05-1. 풀이 및 답 - C언어

#include <stdio.h>

int main(){
    int n, c, x; // 명령의 수, 명령, 추가할 정수
    int size = -1; // 스택의 사이즈
    int stack[1000000];
    scanf("%d", &n);
    
    for (int i=0; i<n; i++) {
        scanf("%d", &c);
        switch (c) {
            case 1:
                scanf("%d", &x);
                stack[++size] = x;
                break;
            case 2:
                if (size >= 0){
                    printf("%d\n", stack[size--]);   
                }
                else {
                    printf("%d\n", -1);
                }
                break;
            case 3:
                printf("%d\n", size+1);
                break;
            case 4:
                if (size == -1) {
                    printf("%d\n", 1);
                }
                else {
                    printf("%d\n", 0);
                }
                break;
            case 5:
                if (size >= 0){
                    printf("%d\n", stack[size]);   
                }
                else {
                    printf("%d\n", -1);
                }
                break;
        }
    }
    return 0;
}

C언어로 스택을 구현해보았다.
입력할 명령의 수를 n, 명령을 c, 스택에 추가해줄 정수를 x로 두었다.
스택의 사이즈는 size로 선언해줬고, 초기값은 -1이다. size가 -1이면 스택이 비어있다는 뜻이다.
입력할 명령의 수 n을 입력받고, i가 0부터 n이 될 때까지 for문을 돌려준다

for문을 돌면서 명령 c를 입력받는다. 
switch 문을 사용해서 각 명령에 따른 출력을 하도록 코드를 작성해줬다.
c가 1이면, x를 입력받고 이 x를 stack의 최상위에 입력해준다. (push)
c가 2면, 스택의 최상위 데이터를 출력하고, 이 데이터를 삭제한다. (pop) 이때, 스택이 비어있는 경우에는 -1을 출력한다.
c가 3이면, 스택에 들어있는 데이터 개수를 출력한다. (size)
c가 4면, 스택이 비어있는 경우엔 "1"을, 아니면 "0"을 출력한다. (empty)
c가 5면, 스택의 최상위 데이터를 출력하고, 이 데이터를 삭제하지 않는다. (top) 이때, 스택이 비어있는 경우에는 -1을 출력한다.

정답!

 

C언어로 스택을 구현하는 것이 조금 까다롭게 느껴졌다.
문제를 다 풀고 다른 분들의 풀이를 서치해보다가 C++은 표준 라이브러리에서 스택 자료 구조를 제공한다는 것을 알게 되었다. ㅎㅎ
C++로 스택을 구현하는 것이 C언어보다 덜 까다로울 것 같아 한번 도전해봤다.

 

05-2. 풀이 및 답 - C++

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

int main(){
    int n, c, x; // 명령의 수, 명령, 추가할 정수
    stack<int> stack;
    scanf("%d", &n);
    
    for (int i=0; i<n; i++) {
        scanf("%d", &c);
        switch (c) {
            case 1:
                scanf("%d", &x);
                stack.push(x);
                break;
            case 2:
                if (!stack.empty()){
                    cout << stack.top() << "\n";
                    stack.pop();
                }
                else {
                    cout << "-1\n";
                }
                break;
            case 3:
                cout << stack.size() << "\n";
                break;
            case 4:
                if (stack.empty()) {
                    cout << "1\n";
                }
                else {
                    cout << "0\n";
                }
                break;
            case 5:
                if (!stack.empty()){
                    cout << stack.top() << "\n";
                }
                else {
                    cout << "-1\n";
                }
                break;
        }
    }
    return 0;
}

기본 함수가 이미 다 만들어져있는 상태에서 가져다 사용하기만 하면 돼서 좀 더 풀기 편했다.
C처럼 index 값을 신경써야하지 않아서 좀 덜 까다롭게 느껴졌다.
물론 C로 코드를 짜본 뒤라 더 수월하게 코드를 작성할 수 있었던 것 같다.

 

역시 정답!

 

 

06. 정리 - C++ stack STL

#include <stack>
stack<int> stack;

stack STL을 사용하기 위해서는 <stack> 헤더 파일을 include 해줘야 한다.
stack은 stack<데이터타입> 이름; 으로 선언해줄 수 있다.

기본 함수는 다음과 같다.
1. push(데이터): 괄호 안 데이터를 스택의 최상위에 추가한다.
2. pop(): 스택의 최상위 데이터를 삭제한다.
3. top(): 스택의 최상위 데이터를 출력한다.
4. size(): 스택의 데이터 개수를 출력한다.
5. empty(): 스택이 비어있는지 여부를 확인한다.