프린터 큐 Printer queue

문제

여러분도 알다시피 여러분의 프린터 기기는 여러분이 인쇄하고자 하는 문서를 인쇄 명령을 받은 ‘순서대로’, 즉 먼저 요청된 것을 먼저 인쇄한다. 여러 개의 문서가 쌓인다면 Queue 자료구조에 쌓여서 FIFO - First In First Out - 에 따라 인쇄가 되게 된다. 하지만 상근이는 새로운 프린터기 내부 소프트웨어를 개발하였는데, 이 프린터기는 다음과 같은 조건에 따라 인쇄를 하게 된다.

현재 Queue의 가장 앞에 있는 문서의 ‘중요도’를 확인한다.

나머지 문서들 중 현재 문서보다 중요도가 높은 문서가 하나라도 있다면, 이 문서를 인쇄하지 않고 Queue의 가장 뒤에 재배치 한다. 그렇지 않다면 바로 인쇄를 한다.

예를 들어 Queue에 4개의 문서(A B C D)가 있고, 중요도가 2 1 4 3 라면 C를 인쇄하고, 다음으로 D를 인쇄하고 A, B를 인쇄하게 된다.

여러분이 할 일은, 현재 Queue에 있는 문서의 수와 중요도가 주어졌을 때, 어떤 한 문서가 몇 번째로 인쇄되는지 알아내는 것이다. 예를 들어 위의 예에서 C문서는 1번째로, A문서는 3번째로 인쇄되게 된다.

입력

첫 줄에 테스트케이스의 수가 주어진다. 각 테스트케이스는 두 줄로 이루어져 있다.

테스트케이스의 첫 번째 줄에는 문서의 개수 N(1 ≤ N ≤ 100)과, 몇 번째로 인쇄되었는지 궁금한 문서가 현재 Queue에서 몇 번째에 놓여 있는지를 나타내는 정수 M(0 ≤ M < N)이 주어진다. 이때 맨 왼쪽은 0번째라고 하자. 두 번째 줄에는 N개 문서의 중요도가 차례대로 주어진다. 중요도는 1 이상 9 이하의 정수이고, 중요도가 같은 문서가 여러 개 있을 수도 있다.

출력

각 테스트 케이스에 대해 문서가 몇 번째로 인쇄되는지 출력한다.

예제 입력 1

3
1 0
5
4 2
1 2 3 4
6 0
1 1 9 1 1 1

예제 출력 1

1
2
5

My solution

이미 프로그래머스의 프린터라는 문제로 풀어본 문제인데 큐를 사용하여 구현할 수 있다.

로직은 틀린 게 없는데 자꾸 값이 이상해서 디버깅해보니 입력이 모두 문자열로 들어오는데 order(몇번째로 출력되는지 궁금한 문서의 순서)가 문자열로 들어와서 if문에서 조건 처리가 안되는 문제였다.

if (order === 0) answerArr.push(answer);

위의 조건문에서 answer들을 모아주는데 order가 "0"으로 입력되고 있었기 떄문이다. 0과 비교하면 false가 나온다.

이렇게 동등 비교 연산자(=)로 구현할 수 있지만 타입이 다를 경우 암묵적으로 타입 변환해주므로 일치 비교 연산자(===)를 사용해서 값과 타입이 둘다 같을 떄 true를 반환하는 것을 권장한다.

입력을 분할하느라 for문이 하나 더 늘어나서 프로그래머스의 편의성을 또 한번 느끼게 되었다.

queue에 문서들이 들어있는 한 반복하면서 맨앞의 문서를 뽑아 그 중요도를 나머지 문서의 중요도와 비교한다.

뽑은 문서가 중요도가 가장 높은 것이 아니라면 뒤로 넣고,

뽑은 문서가 중요도가 가장 높다면 else문으로 넘어가 answer를 먼저 더해준다.(그게 원하는 문서가 아니더라도 answer를 0으로 초기화해줬기 때문에 동일하게 먼저 1을 더해준다. 순서는 1부터 시작하므로) 그리고 그게 원하는 문서라면(order===0) 그동안 누적한 answer가 출력될 것이다. 원하는 문서가 아니면 queue의 문서를 꺼내서 확인하는 작업을 반복한다. 매번 큐에서 문서를 꺼내므로 order를 while문 마지막에 조절해주는 것은 필수다.

소스코드

function solution(i) {
    let input = i.split("\n");
    [t, ...arr] = input;

    let order = 0;
    let queue = [];
    let answer = 0;
    let answerArr = [];
    for (let i = 1; i <= arr.length; i++) {
        if (i % 2 !== 0) {
            queue = 0;
            [_, order] = arr[i - 1].split(" ").map((v) => parseInt(v)); // order 정수화해야 정상작동
        } else {
            queue = arr[i - 1].split(" ").map((v) => parseInt(v)); // 입력을 우선순위변수와 큐로 분리
            while (queue.length) {
                let org = queue.shift();

                // 우선순위 큰 게 존재하면 queue의 뒤로 옮김
                if (queue.filter((v) => v > org).length > 0) queue.push(org);
                else {
                    answer++;
                    if (order === 0) {
                        answerArr += `${answer}\n`;
                        break;
                    }
                    // console.log("answer추가", order, answer, answerArr);
                }
                order--;
                if (order === -1) order = queue.length - 1;
            }
            answer = 0;
        }
    }
    console.log(answerArr);

    return answerArr;
}

test("solution", () => {
    expect(solution("3\n1 0\n5\n4 2\n1 2 3 4\n6 0\n1 1 9 1 1 1\n")).toBe(
        "1\n2\n5\n"
    );
});