백준(BOJ) 10828번 스택
문제 : https://www.acmicpc.net/problem/10828
문제
정수를 저장하는 스택을 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램을 작성하시오.
명령은 총 다섯 가지이다.
- push X: 정수 X를 스택에 넣는 연산이다.
- pop: 스택에서 가장 위에 있는 정수를 빼고, 그 수를 출력한다. 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.
- size: 스택에 들어있는 정수의 개수를 출력한다.
- empty: 스택이 비어있으면 1, 아니면 0을 출력한다.
- top: 스택의 가장 위에 있는 정수를 출력한다. 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.
나의 풀이:
스택의 성질인 FILO 을 기억하면서 풀어주면 되겠다. top을 -1로 두고 push일때는 증가, pop일때는 감소하면서 진행해주자
STL의 stack을 사용해서 풀수도 있고 문제가 간단하기에 그냥 풀어도 된다.
코드 ( C++ )
//STL stack 사용
#include <iostream>
#include <vector>
#include <string>
#include <stack>
using namespace std;
int N;
int main()
{
cin >> N;
stack<int> st;
for (int i = 0; i < N; ++i)
{
string s;
cin >> s;
if (s == "push") {
int n;
cin >> n;
st.push(n);
}
else if (s == "pop"){
if (st.empty())
cout << -1 << "\n";
else {
cout << st.top() << "\n";
st.pop();
}
}
else if (s == "top") {
if (st.empty())
cout << -1 << "\n";
else {
cout << st.top() << "\n";
}
}
else if (s == "size") {
cout << st.size() << "\n";
}
else if (s == "empty") {
if (st.empty())
cout << 1 << "\n";
else
cout << 0 << "\n";
}
}
}
// 배열을 이용하여 풀기
/*
int main()
{
int N;
cin >> N;
int stack[MAX];
int top = -1;
for (int i = 0; i < N; ++i)
{
string s;
cin >> s;
if (s == "push") {
int n;
cin >> n;
stack[++top] = n;
}
else if (s == "top") {
if (top == -1)
cout << -1 << "\n";
else {
cout << stack[top] << "\n";
}
}
else if (s == "size") {
cout << top + 1 << "\n";
}
else if (s == "empty") {
if (top == -1)
cout << 1 << "\n";
else
cout << 0 << "\n";
}
else if (s == "pop") {
if (top == -1)
cout << -1 << "\n";
else {
cout << stack[top--] << "\n";
}
}
}
return 0;
} */
