BOJ 10828 : 스택

문제

---

 

문제 풀어보기

 

 

풀이

---

종종 면접 문제로 나오는 유형이네요.

면접에서는 보통 배열로 구현하라고 합니다.

그러면 벡터의 기본 함수를 사용 할 수 없어서

조금 복잡해지겠네요.

 

 

코드

---

#include <iostream>
#include <string>
#include <vector>

using namespace std;


int main() {
	ios::sync_with_stdio(false);
	cin.tie(NULL);

	vector<int> v;
	int t;
	cin >> t;
	int x;
	string s;
	

	while (t--) {
		cin >> s;
		if (s == "push") cin >> x;

		if (s == "push") v.push_back(x);
		else if (s == "pop") {
			if (v.empty() == true) {
				cout << -1 << '\n';
				continue;
			}
			int temp = v.back();
			v.pop_back();
			cout << temp << '\n';
		}
		else if (s == "size") cout << v.size() << '\n';
		else if (s == "empty") cout << v.empty() << '\n';
		else if (s == "top") {
			if (v.empty() == true) cout << -1 << '\n';
			else cout << v.back() << '\n';
		}

	}


	return 0;
}