[백준] Baekjoon Online Judge
문제
정수를 저장하는 스택을 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램을 작성하시오.
명령은 총 다섯 가지이다.
- 1 X: 정수 X를 스택에 넣는다. (1 ≤ X ≤ 100,000)
- 2: 스택에 정수가 있다면 맨 위의 정수를 빼고 출력한다. 없다면 -1을 대신 출력한다.
- 3: 스택에 들어있는 정수의 개수를 출력한다.
- 4: 스택이 비어있으면 1, 아니면 0을 출력한다.
- 5: 스택에 정수가 있다면 맨 위의 정수를 출력한다. 없다면 -1을 대신 출력한다.
입력
첫째 줄에 명령의 수 N이 주어진다. (1 ≤ N ≤ 1,000,000)
둘째 줄부터 N개 줄에 명령이 하나씩 주어진다.
출력을 요구하는 명령은 하나 이상 주어진다.
출력
출력을 요구하는 명령이 주어질 때마다 명령의 결과를 한 줄에 하나씩 출력한다.
풀이
#include <iostream>
#include <stack>
using namespace std;
stack<int> st;
int main() {
ios::sync_with_stdio(false);
cin.tie(0);
int N;
cin >> N;
while (N--) {
int n;
cin >> n;
if (n == 1) {
int x;
cin >> x;
st.push(x);
}
else if (n == 2) {
if (st.empty()) cout << "-1\n";
else {
cout << st.top() << '\n';
st.pop();
}
}
else if (n == 3) {
cout << st.size() << '\n';
}
else if (n == 4) {
cout << (int)st.empty() << '\n';
}
else {
if (st.empty()) cout << "-1\n";
else cout << st.top() << '\n';
}
}
return 0;
}
<stack> 라이브러리 함수를 사용하여 문제를 풀었다.
'백준 > C++' 카테고리의 다른 글
[Baekjoon/C++] 28279번 - 덱 2 (0) | 2025.01.05 |
---|---|
[Baekjoon/C++] 12789번 - 도키도키 간식드리미 (1) | 2024.12.26 |
[Baekjoon/C++] 17598번 - Animal King Election (1) | 2024.12.18 |
[Baekjoon/C++] 32775번 - 가희와 4시간의 벽 1 (0) | 2024.12.10 |
[Baekjoon/C++] 31306번 - Is Y a Vowel? (0) | 2024.12.08 |