[BOJ] 백준 10845번 큐

링크

https://www.acmicpc.net/problem/10845

문제

정수를 저장하는 큐를 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램을 작성하시오.

명령은 총 여섯 가지이다.

push X: 정수 X를 큐에 넣는 연산이다.
pop: 큐에서 가장 앞에 있는 정수를 빼고, 그 수를 출력한다. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력한다.
size: 큐에 들어있는 정수의 개수를 출력한다.
empty: 큐가 비어있으면 1, 아니면 0을 출력한다.
front: 큐의 가장 앞에 있는 정수를 출력한다. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력한다.
back: 큐의 가장 뒤에 있는 정수를 출력한다. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력한다.

입력

첫째 줄에 주어지는 명령의 수 N (1 ≤ N ≤ 10,000)이 주어진다. 둘째 줄부터 N개의 줄에는 명령이 하나씩 주어진다. 주어지는 정수는 1보다 크거나 같고, 100,000보다 작거나 같다. 문제에 나와있지 않은 명령이 주어지는 경우는 없다.

출력

출력해야하는 명령이 주어질 때마다, 한 줄에 하나씩 출력한다.

풀이

• STL queue를 이용한 풀이

 #include <iostream>
#include <queue>
using namespace std;

int main(void) {
	ios::sync_with_stdio(0);
	cin.tie(0);
	queue<int> q;
	
	int n; 
	cin >> n; 
	while (n--) {
		string s;
		cin >> s;
		if (s == "push") {
			int a;
			cin >> a;
			q.push(a);
		}
		else if (s == "front") {
			if (!q.empty())
				cout << q.front() << '\n';
			else cout << -1 << '\n';
		}
		else if (s == "back") {
			if (!q.empty())
				cout << q.back() << '\n';
			else cout << -1 << '\n';
		}
		else if (s == "size") {
			cout << q.size() << '\n';
		}
		else if (s == "empty") {
			if (q.empty()) cout << 1 << '\n';
			else cout << 0 << '\n';
		}
		else if (s == "pop") {
			if (q.empty()) cout << -1 << '\n';
			else {
				cout << q.front() << '\n';
				q.pop();
			}
		}
	}
}

• 배열을 이용한 queue 직접 구현

#include <iostream>
#include <queue>
using namespace std;

const int mx = 10005;
int q[mx] = {};
int head = 0, tail = 0;

void push(int a) {
	q[tail++] = a;
}

int front() {
	return q[head];
}

int back() {
	return q[tail-1];
}

int size() {
	return tail - head;
}

void pop() {
	head++;
}
int main(void) {
	ios::sync_with_stdio(0);
	cin.tie(0);
	
	int n; 
	cin >> n; 
	while (n--) {
		string s;
		cin >> s;
		if (s == "push") {
			int a;
			cin >> a;
			push(a);
		}
		else if (s == "front") {
			if (head == tail)
				cout << -1 << '\n';
			else
				cout << front() << '\n';
		}
		else if (s == "back") {
			if (head == tail)
				cout << -1 << '\n';
			else
				cout << back() << '\n';
		}
		else if (s == "size") {
			cout << size() << '\n';
		}
		else if (s == "empty") {
			cout << (tail==head) << '\n';
		}
		else if (s == "pop") {
			if (tail == head) 
				cout << -1 << '\n';
			else {
				cout << front() << '\n';
				pop();
			}
		}
	}
}

설명