[C++] BOJ #10828 스택
Posted on 2022. 01. 31
문제
정수를 저장하는 스택을 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램을 작성하시오.
명령은 총 다섯 가지이다.
push X: 정수 X를 스택에 넣는 연산이다.pop: 스택에서 가장 위에 있는 정수를 빼고, 그 수를 출력한다. 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.size: 스택에 들어있는 정수의 개수를 출력한다.empty: 스택이 비어있으면 1, 아니면 0을 출력한다.top: 스택의 가장 위에 있는 정수를 출력한다. 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.
입력
- 첫째 줄에 주어지는 명령의 수 N (1 ≤ N ≤ 10,000)이 주어진다. 둘째 줄부터 N개의 줄에는 명령이 하나씩 주어진다. 주어지는 정수는 1보다 크거나 같고, 100,000보다 작거나 같다. 문제에 나와있지 않은 명령이 주어지는 경우는 없다.
출력
- 출력해야하는 명령이 주어질 때마다, 한 줄에 하나씩 출력한다.
1. 해결방법 및 소스코드
- C++ STL에 내장되어 있는 stack을 활용하였다.
push X명령어는cin을 통해 입력받았다. 연산자>>를 사용하여cin에서 문자열을 추출할 때>>는 첫 번째 공백까지만 반환하고 다른 모든 문자는cin에 남겨져 다음 추출을 기다리는 특성을 활용하였다.
따라서,push X를cin으로 받으면push따로,X따로 변수에 담을 수 있게 되는 것이다.
#include <iostream>
#include <stack>
#include <string>
using namespace std;
int main()
{
//temp: push 명령으로 받을 정수
//command: 명령문자열
int N, temp;
string command;
stack<int> stack;
//N 입력
scanf("%d", &N);
//명령어 입력받고 명령 수행하는 반복문
for (int i = 0; i < N; i++)
{
//명령어 입력
cin >> command;
if (command == "push")
{
cin >> temp;
stack.push(temp);
}
else if (command == "pop")
{
if (stack.empty())
{
printf("-1\n");
}
else
{
printf("%d\n", stack.top());
stack.pop();
}
}
else if (command == "size")
{
printf("%d\n", stack.size());
}
else if (command == "empty")
{
if (stack.empty())
{
printf("1\n");
}
else
{
printf("0\n");
}
}
else if (command == "top")
{
if (stack.empty())
{
printf("-1\n");
}
else
{
printf("%d\n", stack.top());
}
}
}
}