알고리즘/Baekjoon

백준 10828: 스택 (C)

개발하는 크롱 2020. 11. 19. 17:18
반응형

문제 링크: www.acmicpc.net/problem/10828

 

10828번: 스택

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

www.acmicpc.net

소스 코드:

#include <stdio.h>
#include <string.h>
#define MAX_STACK_SIZE 10000

int stack[MAX_STACK_SIZE]; // 스택 배열
int topIndex = -1; // 스택의 top의 인덱스

// push X: 정수 X를 스택에 넣는 연산이다.
void push(int x) {
    stack[++topIndex] = x;
}

// pop: 스택에서 가장 위에 있는 정수를 빼고, 그 수를 출력한다. 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.
int pop() {
    if (empty()) {
        return -1; // 스택이 비었을 경우 -1 출력
    }
    return stack[topIndex--];
}

// size: 스택에 들어있는 정수의 개수를 출력한다.
int size() {
    return topIndex + 1; // 1 더해주는 이유? 배열 인덱스가 0부터 시작하므로. top이 2면 0, 1, 2에 값 있는 것.
}

// empty: 스택이 비어있으면 1, 아니면 0을 출력한다.
int empty() {
    if (topIndex == -1) {
        return 1;
    }
    else return 0;
}

// top: 스택의 가장 위에 있는 정수를 출력한다. 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.
int top() {
    if (topIndex == -1) {
        return -1;
    }
    return stack[topIndex];
}

int main() {
    int N;
    scanf("%d", &N);

    for (int i = 0; i < N; i++) {
        char command[6];
        scanf("%s", command);

        if (!strcmp(command, "push")) {
            int x;
            scanf("%d\n", &x);
            push(x);
        }
        else if (!strcmp(command, "pop")) {
            printf("%d\n", pop());
        }
        else if (!strcmp(command, "size")) {
            printf("%d\n", size());
        }
        else if (!strcmp(command, "empty")) {
            printf("%d\n", empty());
        }
        else if (!strcmp(command, "top")) {
            printf("%d\n", top());
        }
    }

    return 0;
}
반응형