[10845번] 큐 queue

큐, 211102

문제

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

명령은 총 여섯 가지이다.

•

push X: 정수 X를 큐에 넣는 연산이다.

•

pop: 큐에서 가장 앞에 있는 정수를 빼고, 그 수를 출력한다. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력한다.

•

size: 큐에 들어있는 정수의 개수를 출력한다.

•

empty: 큐가 비어있으면 1, 아니면 0을 출력한다.

•

front: 큐의 가장 앞에 있는 정수를 출력한다. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력한다.

•

back: 큐의 가장 뒤에 있는 정수를 출력한다. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력한다.

입력

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

출력

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

예제 입력 1

15
push 1
push 2
front
back
size
empty
pop
pop
pop
size
empty
pop
push 3
empty
front
Plain Text
복사

예제 출력 1

Plain Text
복사

My solution

큐를 구현하는 기초 문제이다. 일반적으로 배열에서 큐에서 하는 연산들을 전부 메서드로 지원해줘서 사용하곤 하는데 여기서 pop 연산은 큐의 dequeue 연산이어서 shift()로 대체했고, push는 그대로 push()를 사용했다.

만약 큐를 이용해서 성능까지 향상해주어야 한다면 네카라쿠배 교육과정에서 배웠던 큐 구현하기를 다시 공부해서 프로토타입 방식으로 큐 객체를 만들어주는 게 좋을 것 같다. 큐 최적화를 해보면서 걸리는 시간도 비교해보았는데 단순히 push와 shift 메서드로 대체하는 것보다 엄청나게 빨랐다.

head와 tail이라는 프로퍼티를 두고 이걸 인덱스를 가리키는 포인터로 사용해서 값을 할당하는 데에 사용했다.

function Queue_2(array) {
    this.array = array ? array : [];
    this.tail = array ? array.length : 0;
    this.head = 0;
}

Queue_2.prototype.enqueue = function (element) {
    return (this.array[this.tail++] = element);
};

Queue_2.prototype.dequeue = function () {
    if (this.tail === this.head) return undefined;

    let element = this.array[this.head];
    delete this.array[this.head++];
    return element;
};
JavaScript
복사

소스코드

function solution(i) {
    let answer = [];
    const input = i.toString().trim().split("\n");
    input.shift();

    let q = [];
    input.forEach((v) => {
        const command = v.split(" ");
        if (command[0] === "pop") answer.push(q.length ? q.shift() : -1);
        else if (command[0] === "size") answer.push(q.length);
        else if (command[0] === "empty") answer.push(q.length ? 0 : 1);
        else if (command[0] === "front") answer.push(q.length ? q[0] : -1);
        else if (command[0] === "back")
            answer.push(q.length ? q[q.length - 1] : -1);
        else if (command[0] === "push") q.push(command[1]);

        console.log("q", q, "ans", answer);
    });

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

    return answer.join("\n");
}

test("solution", () => {
    expect(
        solution(`15
push 1
push 2
front
back
size
empty
pop
pop
pop
size
empty
pop
push 3
empty
front`)
    ).toStrictEqual(`1
2
2
0
1
2
-1
0
1
-1
0
3`);
});
JavaScript
복사