[ C / C++ ] 백준 10828 스택

문제

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

명령은 총 다섯 가지이다.

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

 

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

int number[100001];
int count = 0;
void push(int num) {
	number[count] = num;
	count++;
}

void pop() {
	if (count != 0) {
		count--;
		printf("%d\n", number[count]);
		number[count] = 0;
	}
	else
		printf("%d\n", -1);

}

void top() {
	if (count != 0)
		printf("%d\n", number[count - 1]);
	else
		printf("%d\n", -1);
}

void size() {
	printf("%d\n", count);
}
void empty() {
    
    if (count != 0) {
		printf("0\n");
	}

    else {
		printf("1\n");
	}
}
int main() {
	int n;
	char stack[10];

	scanf("%d", &n); 
	
	int number[100]; 

	for (int i = 0; i < n; i++) {
		scanf("%s", &stack);
		if (strcmp(stack, "push") == 0) {
			int num;
			scanf("%d", &num);
			push(num);
		}
		else if (strcmp(stack, "pop") == 0) {
			pop();
		}
		else if (strcmp(stack, "top") == 0) {
			top();
		}
		else if (strcmp(stack, "size") == 0) {
			size();
		}
		else if (strcmp(stack, "empty") == 0) {
			empty();
		}
	}
}