10845번: 큐

문제:

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

명령은 총 여섯 가지이다.

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

입력:

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

풀이:

• 이 문제는 이전에 풀었던 #10828: 스택을 참고하면 편하다. 거의 같지만 front와 back function이 생겼으며(back은 top과 비슷하긴 하다) pop할 때 제일 뒤가 아닌 앞에서 빼는 방식으로 바꾸어주어야 한다.
• pop 할 때 추가 된 것이 for loop이다 이는 앞에 값이 빠져나가면 그 뒤에 값들이 하나씩 앞으로 밀리게 한다. 또한 if 조건문이 추가 된 이유는 나중에 다 비웠을 때 제일 마지막에 있는 값까지 지워주기 위함이다.

코드:

#include <iostream>

using namespace std;

int n, queue[10000] = {}, back_idx = -1;

void push(int number){
	for(int i = 0; i < n; i++){
		if(queue[i]==0){
			queue[i] = number;
			back_idx = i;
			break;
		}
		else continue;
	}
	return;
}

void pop(){
	if(back_idx == -1) cout << -1 << "\\n";
	else{
		cout<< queue[0] << "\\n";
		//pop한 이후 앞으로 값들을 하나씩 땡겨오는 반복문
		for(int i = 0; i < back_idx; i++){
			queue[i] = queue[i+1];
			queue[i+1] = 0;
		}
		back_idx--;
		//비웠을 경우에 마지막 남은 값까지 0으로 처리
		if(back_idx == -1) queue[0] = 0;
	}
	return;
}

void size(){
	cout << back_idx+1 << "\\n";
	return;
}

void empty(){
	if(back_idx == -1) cout << 1 <<"\\n";
	else cout << 0 <<"\\n";
	return;
}

void front(){
	if(back_idx == -1) cout << -1 <<"\\n";
	else{
		cout<< queue[0] << "\\n";
	}
	return;
}

void back(){
	if(back_idx == -1) cout << -1 <<"\\n";
	else{
		cout<< queue[back_idx] << "\\n";
	}
	return;
}

int main(void){
	cin >> n;
	for(int i = 0; i < n; i++){
		string temp;
		cin >> temp;
		if(temp == "push"){
			int number;
			cin >> number;
			push(number);
		}
		else if(temp == "pop") pop();
		else if(temp == "size") size();
		else if(temp == "empty") empty();
		else if(temp == "front") front();
		else if(temp == "back") back();
	}
	
}