백준 알고리즘 10828번 스택 (C언어)

실버Ⅳ

 

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

10828번: 스택

 

시간 제한	메모리 제한	제출	정답	맞힌 사람	정답 비율
0.5 초 (추가 시간 없음)	256 MB	209170	74003	53986	37.313%

 

문제

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

명령은 총 다섯 가지이다.

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

입력

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

출력

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

예제 입력 1

14
push 1
push 2
top
size
empty
pop
pop
pop
size
empty
pop
push 3
empty
top

예제 출력 1

2
2
0
2
1
-1
0
1
-1
0
3

예제 입력 2

7
pop
top
push 123
top
pop
top
pop

예제 출력 2 

-1
-1
123
123
-1
-1

반응형

풀이 방법

이 문제는 스택의 개념을 이해한 후, 스택의 push와 pop을 비롯해 이 문제에서 요구하는 size, empty, top의 기능들을 구현하면 되는 문제입니다. 이 문제의 경우 배열을 이용해 스택을 구현해 주었습니다.

 

push : push의 경우 top이 stack의 max_size-1보다 크거나 같을 경우에 예외처리를 해줍니다. top이 max_size-1보다 크거나 같지 않을 경우 top의 값을 1 증가시킨 후 stack의 top번째 인덱스에 값을 추가해 줍니다.

 

pop : pop의 경우 top이 -1일 경우에 예외처리를 해줍니다. top이 -1이 아닐 경우 stack의 top번째 인덱스에 들어있는 값을 리턴해 준 후, top의 값을 1 감소 시킵니다.

 

size : size의 경우 top+1의 값을 print 해줍니다.

 

empty : empty의 경우 top의 값이 -1일 경우 1, 아닐 경우 0을 출력해줍니다. 

 

Top : top의 값이 -1일 경우 -1, 아닐 경우 stack의 top번째 인덱스에 있는 값을 출력해줍니다.

 

내 코드

#pragma warning(disable:4996)
#include <stdio.h>
#include <string.h>
 
#define MAX_SIZE 100000
 
void push(int stack[], int x, int *top);
int pop(int stack[], int *top);
void size(int *top);
void empty(int *top);
void Top(int stack[], int *top);
 
int main() {
    int n, top = -1, in, num = 0;
    int stack[MAX_SIZE];
    char str[6];
 
    scanf("%d", &n);
    for (int i = 0; i < n; i++) {
        scanf("%s", str);
        if (strcmp(str, "push") == 0) {
            scanf("%d", &in);
            push(stack, in, &top);
        }
        else if (strcmp(str, "pop") == 0) {
            num = pop(stack, &top);
            printf("%d\n", num);
        }
        else if (strcmp(str, "size") == 0) {
            size(&top);
        }
        else if (strcmp(str, "empty") == 0) {
            empty(&top);
        }
        else if (strcmp(str, "top") == 0) {
            Top(stack, &top);
        }
        else
            return -1;
    }
 
    return 0;
}
 
void push(int stack[], int x, int *top) {
    if (*top >= MAX_SIZE - 1)
        return;
    else
        stack[++(*top)] = x;
}
 
int pop(int stack[], int *top) {
    if (*top == -1)
        return -1;
    else
        return stack[(*top)--];
}
 
void size(int *top) {
    printf("%d\n", *top + 1);
}
 
void empty(int *top) {
    int n;
    n = (*top == -1) ? 1 : 0;
    printf("%d\n", n);
}
 
void Top(int stack[], int *top) {
    int n;
    n = (*top == -1) ? -1 : stack[*top];
    printf("%d\n", n);
}

 

코드설명

int n, top = -1, in, num = 0;

스택의 경우 top의 값은 -1부터 시작합니다.

 

for (int i = 0; i < n; i++) {
    scanf("%s", str);
    if (strcmp(str, "push") == 0) {
        scanf("%d", &in);
        push(stack, in, &top);
    }
    else if (strcmp(str, "pop") == 0) {
        num = pop(stack, &top);
        printf("%d\n", num);
    }
    else if (strcmp(str, "size") == 0) {
        size(&top);
    }
    else if (strcmp(str, "empty") == 0) {
        empty(&top);
    }
    else if (strcmp(str, "top") == 0) {
        Top(stack, &top);
    }
    else
        return -1;
}

문자열 비교로 입력에 따라 그게 맞는 함수를 실행하도록 하는 부분입니다.

 

*top

top의 값은 global 변수로 선언하지 않고, pointer을 이용해 값을 어느 함수에서나 변경할 수 있도록 했습니다.