백준 10828_스택(C++)

 

www.acmicpc.net/problem/10828

10828번: 스택

[문제 풀이]

STL을 이용한 것과 직접 구현한 것 두 가지로 문제를 풀어보았습니다.

 

이 문제는 단순히 스택을 구현하는 것이므로 설명은 생략하겠습니다.

코드는 STL을 이용한 코드와 직접 구현한 코드 두 가지를 적어 놓은 것입니다.

[STL 코드]

#include<iostream>
#include<stack>
#include<string>

using namespace std;

int N;

stack<int> stac;

int main() {

	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	cout.tie(NULL);

	cin >> N;

	string exec;

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

		if (exec == "push") {    //push
			int temp;
			cin >> temp;
			stac.push(temp);
		}
		else if (exec == "pop") {    //pop
			if (stac.empty())
				cout << -1 << endl;
			else {
				cout << stac.top() << endl;
				stac.pop();
			}
		}
		else if (exec == "size") {    //size
			cout << stac.size() << endl;
		}
		else if (exec == "empty") {    //empty
			if (stac.empty())
				cout << 1 << endl;
			else
				cout << 0 << endl;
		}
		else if (exec == "top") {     //top
			if (stac.empty())
				cout << -1 << endl;
			else
				cout << stac.top() << endl;
		}
	}

	return 0;
}

[직접 구현 코드]

#include<iostream>
#include<string>

using namespace std;

int arr[10000] = { 0, };

int main() {

	ios::sync_with_stdio(false);
	cin.tie(null);

	int n; cin >> n;

	int idx = -1;

	for (int i = 0; i < n; i++) {
		string str;

		cin >> str;

		if (str == "push") {
			int x; cin >> x;
			idx++;
			arr[idx] = x;
		}
		else if (str == "pop") {

			if (idx == -1)
				cout << "-1" << endl;
			else {
				cout << arr[idx] << endl;
				arr[idx] = 0;
				idx--;
			}
		}
		else if (str == "size") {
			if (idx == -1)
				cout << 0 << endl;
			else
				cout << idx + 1 << endl;
		}
		else if (str == "empty") {
			if (idx == -1)
				cout << 1 << endl;
			else
				cout << 0 << endl;
		}
		else if (str == "top") {
			if (idx == -1)
				cout << -1 << endl;
			else
				cout << arr[idx] << endl;
		}

	}


	return 0;
}