오늘의 코딩 2일차

• 사용 언어 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 <stack>
#include <string>
using namespace std;

int main()
{
	int n;
	cin >> n;

	stack<int> stack;
	string str;

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

		if (str == "push")
		{
			int num;
			cin >> num;
			stack.push(num);
		}
		else if (str == "pop")
		{
			if (!stack.empty())
			{
				cout << stack.top() << endl;
				stack.pop();
			}
			else
			{
				cout << "-1" << endl;
			}
		}
		else if (str == "size")
		{
			cout << stack.size() << endl;
		}
		else if (str == "empty")
		{
			if (!stack.empty())
			{
				cout << "0" << endl;
			}
			else
			{
				cout << "1" << endl;
			}
		}
		else if (str == "top")
		{
			if (!stack.empty())
			{
				cout << stack.top() << endl;
			}
			else
			{
				cout << "-1" << endl;
			}
		}
	}
	return 0;
}

C++의 stack 기능을 이용하여 코드를 작성.

첫 정수형 n은 명령의 수를 입력해준다.

 

문자열 값을 입력받아 조건을 구분하게 해준다.

push X 는 push를 입력하고 나서 정수형을 입력하면 해당 값을 스택에 넣는다.

pop은 스택에 넣어진 값의 맨 처음 값을 불러오기 위해 stack.top으로 최상위 데이터 값을 불러온다. 그 뒤 stack.pop()으로 해당 값을 삭제해준다.

size는 말 그대로 stack의 현재 사이즈를 물어보는 것이기에 stack.size()를 불러와준다.

empty는 stack.empty()로 스택이 비어있는지 확인해준다.

top은 스택의 최상위 데이터를 불러오기 위해 stack.top()으로 불러와준다.

 

스택에 대해 공부하면서 그걸 활용해 풀어보기 좋은 문제이다.