[Baekjoon/C++] 10828번 - 스택

Baekjoon Online Judge

문제로 이동

 

문제

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

명령은 총 다섯 가지이다.

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

 

입력

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

 

출력

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

 

---

예제 입력	예제 출력
14 push 1 push 2 top size empty pop pop pop size empty pop push 3 empty top	2 2 0 2 1 -1 0 1 -1 0 3
7 pop top push 123 top pop top pop	-1 -1 123 123 -1 -1

---

풀이

#include <iostream>
using namespace std;

struct stack {
	int top = -1;
	int element[10001] = {};
};

stack s;

// 정수를 스택에 넣는 연산
void push(int n) {
	s.element[++s.top] = n;
}

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

// 스택에서 가장 위에 있는 정수를 빼고, 그 수를 출력
int pop() {
	if (empty())
		return -1;
	else
		return s.element[s.top--];
}

// 스택에 들어있는 정수의 개수를 출력
int size() {
	return s.top + 1;
}

// 스택의 가장 위에 있는 정수를 출력
int top() {
	return s.element[s.top];
}

int main() {
	int N;
	cin >> N;

	for (int i = 0; i < N; i++) {
		string s;
		cin >> s;

		int num;

		if (s == "push") {
			cin >> num;
			push(num);
		}
		else {
			if (s == "pop")
				num = pop();
			else if (s == "size")
				num = size();
			else if (s == "empty")
				num = empty();
			else
				num = top();

			// 출력
			printf("%d\n", num);
		}
	}

	return 0;
}