(c++)백준 10828번: 스택

https://www.acmicpc.net/problem/10828

10828번: 스택

[풀이]

C++ STL의 stack을 사용하여 풀 수 있다.

순서대로 명령을 입력받자마자 그 명령에 해당하는 일을 처리한다.

 

#include <iostream>
#include <stack>
#include <string>
using namespace std;

void solution() {
	stack<int>s;
	int N, pushNum;
	string str;
	cin >> N;
	for (int i = 0; i < N; i++) {
		cin >> str;
		if (str == "push") {
			//pushNum을 스택에 넣기
			cin >> pushNum;
			s.push(pushNum);
		}
		else if (str == "pop") {
			//스택에 들어있는 정수가 없는 경우 -1 출력
			if (s.empty()) {
				cout << -1 << endl;
				continue;
			}
			//스택 가장 위에 있는 정수 빼고, 그 수 출력
			int Top = s.top();
			cout << Top << endl;
			s.pop();
		}
		else if (str == "size") {
			//스택에 들어있는 정수의 개수 출력
			cout << s.size() << endl;
		}
		else if (str == "empty") {
			//스택이 비어있으면 1, 아니면 0 출력
			if (s.empty())
				cout << 1 << endl;
			else
				cout << 0 << endl;
		}
		else if (str == "top") {
			//스택에 들어있는 정수가 없는 경우 -1 출력
			if (s.empty()) {
				cout << -1 << endl;
				continue;
			}
			//스택의 가장 위에 있는 정수 출력
			cout << s.top() << endl;
		}
	}
}

int main() {
	solution();
	return 0;
}