[백준 18258번] 큐 2
J-Shine
[백준 18258번] 큐 2 C++ 풀이
시간 제한
1초
메모리 제한
256MB
문제
정수를 저장하는 큐를 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램을 작성하시오.
명령은 총 여섯 가지이다.
push X: 정수 X를 큐에 넣는 연산이다.
pop: 큐에서 가장 앞에 있는 정수를 빼고, 그 수를 출력한다. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력한다.
size: 큐에 들어있는 정수의 개수를 출력한다.
empty: 큐가 비어있으면 1, 아니면 0을 출력한다.
front: 큐의 가장 앞에 있는 정수를 출력한다. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력한다.
back: 큐의 가장 뒤에 있는 정수를 출력한다. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력한다.
입력
첫째 줄에 주어지는 명령의 수 N (1 ≤ N ≤ 2,000,000)이 주어진다. 둘째 줄부터 N개의 줄에는 명령이 하나씩 주어진다.
주어지는 정수는 1보다 크거나 같고, 100,000보다 작거나 같다. 문제에 나와있지 않은 명령이 주어지는 경우는 없다.
출력
출력해야하는 명령이 주어질 때마다, 한 줄에 하나씩 출력한다.
STL queue를 이용한 내 풀이
#include <bits/stdc++.h>
using namespace std;
int main() {
ios_base::sync_with_stdio(0);
cin.tie(0);
queue<int> q;
int n(0), m(0);
string command;
cin >> n;
for (int i = 0; i < n; i++) {
cin >> command;
if (command == "push") {
cin >> m;
q.push(m);
}
else if (command == "pop") {
if (!q.empty()) {
cout << q.front() << "\n";
q.pop();
}
else
cout << "-1" << "\n";
}
else if (command == "size") {
cout << q.size() << "\n";
}
else if (command == "empty") {
if(!q.empty()) cout << 0 << "\n";
else cout << 1 << "\n";
}
else if (command == "front") {
if (!q.empty()) cout << q.front() << "\n";
else cout << "-1" << "\n";
}
else if (command == "back") {
if (!q.empty()) cout << q.back() << "\n";
else cout << "-1" << "\n";
}
}
}
STL queue를 사용하여보았다.
queue를 구현한 내 풀이
#include <bits/stdc++.h>
using namespace std;
int empt(int& head, int& tail) {
if (head == tail) return 1;
else return 0;
}
void push(int arr[], int& tail, int x) {
arr[tail++] = x;
}
void pop(int arr[], int& head, int& tail) {
if(!empt(head, tail)) cout << arr[head++] << "\n";
else cout << -1 << "\n";
}
void size(int& head, int& tail) {
cout << tail - head << "\n";
}
void front(int arr[], int& head, int& tail) {
if (!empt(head, tail)) cout << arr[head] << "\n";
else cout << -1 << "\n";
}
void back(int arr[], int& head, int& tail) {
if (!empt(head, tail)) cout << arr[tail - 1] << "\n";
else cout << -1 << "\n";
}
int main() {
ios_base::sync_with_stdio(0);
cin.tie(0);
int n(0), m(0), q[2000005], head(0), tail(0);
string command;
cin >> n;
for (int i = 0; i < n; i++) {
cin >> command;
if (command == "push") {
cin >> m;
push(q, tail, m);
}
else if (command == "pop") {
pop(q, head, tail);
}
else if (command == "size") {
size(head, tail);
}
else if (command == "empty") {
cout << empt(head, tail) << "\n";
}
else if (command == "front") {
front(q, head, tail);
}
else if (command == "back") {
back(q, head, tail);
}
}
}
10845번 큐 문제와 동일하고 명령의 수가 200만개로 늘어났다.
코드는 그냥 복붙 후 배열의 수 200만5개로 바꿔줬음