#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();
}
}
Leave a comment