[백준/C++] 10828번 풀이

업데이트:



문제 정보


풀이

문제

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

명령은 총 다섯 가지이다.

  • push X: 정수 X를 스택에 넣는 연산이다.

  • pop: 스택에서 가장 위에 있는 정수를 빼고, 그 수를 출력한다. 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.

  • size: 스택에 들어있는 정수의 개수를 출력한다.

  • empty: 스택이 비어있으면 1, 아니면 0을 출력한다.

  • top: 스택의 가장 위에 있는 정수를 출력한다. 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.

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

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

코드

#include "iostream"
#include "string"

#define STACK_SIZE 10000

using namespace std;

class Stack {
private:
	int stack[STACK_SIZE];
	int size = 0;

public:
	Stack() {}

	void push(int n) {
		stack[size++] = n;
	}

	int pop() {
		return size > 0 ? stack[--size] : -1;
	}

	int getSize() {
		return size;
	}

	int empty() {
		return size != 0 ? 0 : 1;
	}

	int top() {
		return size > 0 ? stack[size - 1] : -1;
	}

	void print() {
		cout << "[";
		for (int i = 0; i < size; i++) {
			cout << stack[i];
		}
		cout << "]" << endl;
	}
};

int main() {
	ios_base::sync_with_stdio(false);
	Stack stack = Stack();

	int n;
	cin >> n;
	cin.ignore(5, '\n');

	for (int i = 0; i < n; i++) {
		string command;
		getline(cin, command);
		
		switch (command.front()) {
		case 'p':
			if (command.at(1) == 'u') {
				stack.push(atoi(command.substr(5).c_str()));
			}
			else {
				cout << stack.pop() << endl;
			}
			break;
		case 's':
			cout << stack.getSize() << endl;
			break;
		case 'e':
			cout << stack.empty() << endl;
			break;
		case 't':
			cout << stack.top() << endl;
			break;
		}
	}

	return 0;
}

설명

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

명령은 총 다섯 가지이다.

  • push X: 정수 X를 스택에 넣는 연산이다.

  • pop: 스택에서 가장 위에 있는 정수를 빼고, 그 수를 출력한다. …를 만족하도록 로직을 구성하는 것입니다.

코드는 입력을 파싱한 뒤 조건 분기와 계산을 순서대로 수행하고, 문제에서 요구한 형식으로 결과를 출력합니다.

경계값과 예외 케이스도 함께 고려해 오답이 나기 쉬운 상황을 방지합니다.



이런 주제는 어떠신가요?

비슷한 난이도와 유형의 문제를 이어서 보면 풀이 감각을 더 빠르게 잡기 좋습니다.

댓글남기기