10828번 스택 BOJ 백준 C언어

문제

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

명령은 총 다섯 가지이다.

• push X: 정수 X를 스택에 넣는 연산이다.
• pop: 스택에서 가장 위에 있는 정수를 빼고, 그 수를 출력한다. 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.
• size: 스택에 들어있는 정수의 개수를 출력한다.
• empty: 스택이 비어있으면 1, 아니면 0을 출력한다.
• top: 스택의 가장 위에 있는 정수를 출력한다. 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.

입력

첫째 줄에 주어지는 명령의 수 N (1 ≤ N ≤ 10,000)이 주어진다. 둘째 줄부터 N개의 줄에는 명령이 하나씩 주어진다. 주어지는 정수는 1보다 크거나 같고, 100,000보다 작거나 같다. 문제에 나와있지 않은 명령이 주어지는 경우는 없다.

출력

출력해야하는 명령이 주어질 때마다, 한 줄에 하나씩 출력한다.

구현

기본적으로 stack 이라는 개념에 대해서 이해해야만 풀 수 있는 문제이다. 스택은 차곡차곡 쌓이는 것에 기반된 자료구조이다. 다음 사진과 같은 방식으로 구현된다. 스택이 어떤 방식으로 구현되는지만 이해한다면 쉽게 풀 수 있는 문제이다.

해당 문제에서는 STL에서 기본적으로 주어지는 stack 자료형을 사용하지 않고 직접 구현해보길 추천하는 것 같아서 배열로 스택을 구현한 후 사용하였다.

 

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <algorithm>
#define _CRT_SECURE_NO_WARNINGS

int main(void) {
    int N, i, j = 0, num;
    int stack[10001] = { 0 };
    char str[6];
    scanf("%d", &N);
    for (i = 0; i < N; i++) {
        scanf("%s", str);
        if (strcmp(str,"push")==0) {
            scanf("%d", &num);
            j++;
            stack[j] = num;
        }
        else if (strcmp(str, "pop") == 0) {
            if (j == 0) {
                printf("-1\n");
            }
            else {
                printf("%d\n", stack[j]);
                stack[j] = 0;
                j--;
            }
        }
        else if (strcmp(str, "size") == 0) {
            printf("%d\n", j);
        }
        else if (strcmp(str, "empty") == 0) {
            if (j == 0) {
                printf("1\n");
            }
            else {
                printf("0\n");
            }
        }
        else if (strcmp(str, "top") == 0) {
            if (j == 0) {
                printf("-1\n");
            }
            else {
                printf("%d\n", stack[j]);
            }
        }
    }
}

정리

프로그래밍 언어에는 아주 다양한 자료구조형이 존재한다. 그 중에서 어떻게 보면 가장 쉽다는 stack의 기본 개념을 파악하고 어떤 원리로 사용되며 어떤 문제에서 해당 자료구조형을 쓸지를 파악하는 것이 가장 중요한 점인것 같다.