백준 10828번 큐

문제

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

명령은 총 여섯 가지이다.

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

입력

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

출력

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

---

풀이/접근 방법(JavaScript)

10828 스택 문제 에서 코드를 살짝만 수정해줬다.

큐는 스택과 달리 pop(원소를 제거)연산이 맨 앞에서 이루어진다. 따라서 자바스크립트의 배열 메서드 중 하나인 shift를 이용해 맨 앞의 요소를 제거하고, 반환해줬다.

코드(JavaScript)

const fs = require('fs');
const input = fs.readFileSync('/dev/stdin').toString().split('\n');

const queue = [];
const answer = [];
const len = Number(input[0]);

for(let i=1; i<=len; i++){
    let cmd = input[i].split(' ');
    switch(cmd[0]) {
        case 'push':
            queue.push(cmd[1]);
            break;
        case 'pop':
            answer.push(queue.shift() || -1);
            break;
        case 'size':
            answer.push(queue.length);
            break;
        case 'empty':
            answer.push(queue[0] ? 0 : 1);
            break;
        case 'front':
            answer.push(queue[0] || -1);
            break;
        case 'back':
            answer.push(queue[queue.length-1] || -1);
            break;
    }
}

console.log(answer.join('\n'));

---

풀이/접근 방법(C++) ver1.

c++의 queue STL을 사용해 풀었다.
#include <queue> 헤더파일 작성해준 뒤 `

코드(C++)

풀이/접근 방법(C++) ver2.

배열로 큐를 직접 구현해 풀었다.
아래 코드에서 f는 큐의 front, r은 rear를 의미한다.

코드(C++)

#include <iostream>

using namespace std;

int queue[10000];
int f = 0, r = 0;

void push(int num){
    queue[r++]=num;
}

int empty(){
    if(r-f == 0) return 1;
    else return 0;
}

int pop(){
    if(!empty()) return queue[f++];
    else return -1;
}

int size(){
    return r-f;
}

int front(){
    if(!empty()) return queue[f];
    else return -1;
}

int back(){
    if(!empty()) return queue[r-1];
    else return -1;
}

int main(){
    ios::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
    
    int n;
    cin >> n;
    for(int i=0; i<n; i++){
        string cmd;
        cin >> cmd;
        if(cmd == "push"){
            int num;
            cin >> num;
            push(num);
        }
        else if(cmd == "pop") cout << pop() << '\n';
        else if(cmd == "size") cout << size() << '\n';
        else if(cmd == "empty") cout << empty() << '\n';
        else if(cmd == "front") cout << front() << '\n';
        else if(cmd == "back") cout << back() << '\n';
    }
    
    return 0;
}