10828번: 스택(BOJ C/C++)

10828번: 스택(BOJ C/C++)

사용 언어: C

 

문제

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

명령은 총 다섯 가지이다.

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

입력

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

출력

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

 

풀이

문제가 어렵지 않고 풀이도 직관적이어서 부가 설명은 필요 없다고 생각해서 더 이상의 설명은 생략하겠다.

enum을 추가해서 enum{PUSH=1,POP,SIZE,EMPTY,TOP}; 이렇게도 써볼까 생각해봤는데 가독성이 오히려 떨어질 것 같아서 생략했다.

 

#include <stdio.h>
#include <string.h>
#define MAX 100001

int Stack[MAX];
int topnum=-1;

void push(int n)
{
	Stack[++topnum]=n;
}

int pop()
{
	if(topnum<0)
	{
		printf("-1\n");
		return 0;
	}
	else
	{
		printf("%d\n",Stack[topnum]);
		return Stack[topnum--];
	}
}

void size()
{
	printf("%d\n",topnum+1);
}

void empty()
{
	if(topnum<0)
	{
		printf("1\n");
	}
	else
		printf("0\n");
}

void top()
{
	if(topnum<0)
	{
		printf("-1\n");
	}
	else
		printf("%d\n",Stack[topnum]);
}

int main(void)
{
	char Get_Com[10];
	int N,pushnum;
	scanf("%d",&N);
	int Stack[N];
	for(int i=0;i<N;i++)
	{
		scanf("%s",Get_Com);
		if(strcmp(Get_Com,"push")==0)
		{
			scanf("%d",&pushnum);
			push(pushnum);
		}
		else if(strcmp(Get_Com,"pop")==0)
			pop();
		else if(strcmp(Get_Com,"size")==0)
			size();
		else if(strcmp(Get_Com,"empty")==0)
			empty();
		else if(strcmp(Get_Com,"top")==0)
			top();
	}
	return 0;
}