[Baekjoon] Class2 31 - 스택 [10828]

스택 [10828번]

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

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

입력

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

출력

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

생각해볼점

스택의 기본 기능을 구현하는 문제다. 하노이 탑 방식의 스택 구조를 이해하고있다면 쉽게 해결할 수 있다.
스택에 대한 내용은 이전에 스택 수열 문제에서 다룬 적이 있다.

코드 구현

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

int main()
{
	int		num, *stack, stack_idx, push_temp;
	char	order[10];
	
	scanf("%d", &num);
	stack = (int *)malloc(num * sizeof(int));

	stack_idx = 0;
	stack[0] = -1;
	while (num-- > 0)
	{
		scanf("%s", order);

		if (strcmp(order, "push") == 0)
		{
			scanf("%d", &push_temp);
			stack[stack_idx++] = push_temp;
		}
		else if (strcmp(order, "pop") == 0)
		{
			if (stack[0] == -1)
				printf("-1\n");
			else
			{
				printf("%d\n", stack[--stack_idx]);
				stack[stack_idx] = -1;
			}
		}
		else if (strcmp(order, "size") == 0)
			printf("%d\n", stack_idx);
		else if (strcmp(order, "empty") == 0)
		{
			if (stack[0] == -1)
				printf("1\n");
			else
				printf("0\n");
		}
		else if (strcmp(order, "top") == 0)
		{
			if (stack[0] == -1)
				printf("-1\n");
			else
				printf("%d\n", stack[stack_idx - 1]);
		}
	}

	return (0);
}