[백준 18258] 큐 2 - C++

문제

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

문제 설명

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

명령은 총 여섯 가지이다.

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

입출력

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

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

풀이

1. 배열로 구현

• 사용한 자료구조 : 배열
• 풀이 내용
  - front : 큐의 제일 앞부분
  • rear : 큐의 제일 뒷부분
    - 초기 상태
    • front : 0
      • rear : 0
  • push 하는 경우
    front는 변하지 않고 rear는 이전보다 1 증가
    • front(t+1) : front(t)
      • rear(t+1) : rear(t) + 1
  • pop 하는 경우
    rear는 변하지 않고 front는 이전보다 1증가
    • front(t+1) : front(t) + 1
    • rear(t+1) : rear(t) + 1
• 코드

#include <iostream>
using namespace std;


const int MX = 100005;
int dat[MX];
int front, rear;
int N;
int main(void) {
	ios::sync_with_stdio(0);
	cin.tie(0);

	string comm;
	

	cin >> N;
	for (int i = 0; i < N; i++) {
		cin >> comm;
		if (comm == "push") {
			int num;
			cin >> num;
			dat[rear++] = num;
		}
		else if (comm == "pop") {
			if ((rear - front) == 0) {
				cout << -1 << '\n';
			}
			else {
				cout << dat[front++] << '\n';
			}
		}
		else if (comm == "size") {
			cout << rear - front << '\n';
		}
		else if (comm == "empty") {
			if ((rear - front) == 0) {
				cout << 1 << '\n';
			}
			else {
				cout << 0 << '\n';
			}
		}
		else if (comm == "front") {
			if ((rear - front) == 0) {
				cout << -1 << '\n';
			}
			else {
				cout << dat[front] << '\n';
			}

		}
		else if (comm == "back") {
			if ((rear - front) == 0) {
				cout << -1 << '\n';
			}
			else {
				cout << dat[rear-1] << '\n';
			}
		}
	}
}

2. STL queue로 구현

• 사용한 자료구조 : 큐
• 풀이 내용
  STL queue 기본 함수들을 이용하여 구현
• 코드

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


int N;
int main(void) {
	ios::sync_with_stdio(0);
	cin.tie(0);

	queue<int> myqueue;
	string comm;


	cin >> N;
	for (int i = 0; i < N; i++) {
		cin >> comm;
		if (comm == "push") {
			int num;
			cin >> num;
			myqueue.push(num);
		}
		else if (comm == "pop") {
			if (myqueue.empty()) {
				cout << -1 << '\n';
			}
			else {
				cout << myqueue.front() << '\n';
				myqueue.pop();
			}
		}
		else if (comm == "size") {
			cout << myqueue.size() << '\n';
		}
		else if (comm == "empty") {
			cout << (int)myqueue.empty() << '\n';
		}
		else if (comm == "front") {
			if (myqueue.empty()) {
				cout << -1 << '\n';
			}
			else {
				cout << myqueue.front() << '\n';
			}
		}
		else if (comm == "back") {
			if (myqueue.empty()) {
				cout << -1 << '\n';
			}
			else {
				cout << myqueue.back() << '\n';
			}
		}
	}
}