[伯俊/C+]10828号:スタック

質問する

入力

しゅつりょく

例

アイデア

コード#コード#

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

int main() {
	int N;	// 반복 횟수
	stack<int> stk;
	cin >> N;
	string cmd;	// 명령어
	while (N--) {
		cin >> cmd;
		if (cmd == "push") {
			int val;
			cin >> val;
			stk.push(val);
		}
		else if (cmd == "pop") {
			if (stk.size() == 0) {
				cout << -1 << endl;
			}
			else {
				cout << stk.top() << endl;
				stk.pop();
			}
		}
		else if (cmd == "size") {
			cout << stk.size() << endl;
		}
		else if (cmd == "empty") {
			if (stk.size() == 0)
				cout << 1 << endl;
			else
				cout << 0 << endl;
		}
		else if (cmd == "top") {
			if (stk.size() == 0) {
				cout << -1 << endl;
			}
			else
				cout << stk.top() << endl;
		}
	}
	return 0;
}

試行錯誤