[Algorithm/Java] 백준 1966번 - 프린터 큐

2025-06-24 4 분 소요

https://www.acmicpc.net/problem/1966

📌 문제

문제 유형

구현
자료 구조
시뮬레이션
큐

문제 설명

여러분도 알다시피 여러분의 프린터 기기는 여러분이 인쇄하고자 하는 문서를 인쇄 명령을 받은 ‘순서대로’, 즉 먼저 요청된 것을 먼저 인쇄한다. 여러 개의 문서가 쌓인다면 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
2
5

🔍 문제 풀이

맨 앞 문서가 가장 중요한 문서일 때:

그 문서가 m번째 문서라면 → 해당 문서를 출력하고 종료
그 문서가 m번째 문서가 아니라면 → 해당 문서를 뒤로 보냄

맨 앞 문서가 가장 중요한 문서가 아닐 때:

뒤로 밀리고, 계속해서 문서를 비교하면서 중요한 문서가 앞으로 오기를 기다림

문서 수(N): 4개
문서 중요도: [2, 1, 4, 3]
알고 싶은 문서 위치(M): 0번째 (즉, 중요도 2인 문서)

문서의 인덱스와 중요도를 함께 저장해야 함 → int[] 로 관리
- M번째에 위치한 문서를 추적하기 위함
중요도가 높은 문서가 뒤에 있으면 현재 문서를 뒤로 보냄
인쇄될 때마다 카운트를 올리고, 내가 요청한 문서(M번째)가 출력되면 정답 출력

Deque<int[]>

Deque<int[]> dq = new ArrayDeque<>(); 를 선언해 [문서 인덱스, 중요도] 형태로 저장한다.

문서를 앞에서 하나 꺼낸 뒤, 덱 안에 더 높은 중요도의 문서가 있는지 확인한다.
- 있으면 → 다시 뒤로 넣는다.
- 없으면 → 출력 카운트를 증가시킨다.
그 문서의 인덱스가 m과 같다면 → 정답으로 출력한다.

시뮬레이션

입력:

문서 중요도: [2, 1, 4, 3]
찾고 싶은 문서 위치(m): 0

문서 저장 (Deque에 삽입):

dq.offerLast(new int[]{0, 2}); // A
dq.offerLast(new int[]{1, 1}); // B
dq.offerLast(new int[]{2, 4}); // C
dq.offerLast(new int[]{3, 3}); // D

→ 덱 초기 상태:

[0,2] [1,1] [2,4] [3,3]

시뮬레이션:

꺼냄: [0,2] → 뒤에 [2,4] 있음 → 중요도 더 높음 → 뒤로 보냄
덱 상태: [1,1] [2,4] [3,3] [0,2]
꺼냄: [1,1] → 뒤에 [2,4] 있음 → 뒤로 보냄
덱 상태: [2,4] [3,3] [0,2] [1,1]
꺼냄: [2,4] → 제일 중요 → 인쇄 printOrder = 1
꺼냄: [3,3] → 제일 중요 → 인쇄 printOrder = 2
꺼냄: [0,2] → 남은 문서들보다 중요도 높음 → 인쇄 printOrder = 3
front[0] == m이므로 → 정답: 3

M번째 문서가 세 번째로 인쇄되므로 정답은 3이다.

⚠️ for-each 안에서 큐 수정 연산X

for-each 안에서 큐 수정이 안 되는 이유

① for-each는 내부적으로 Iterator를 사용

for-each는 컬렉션을 순회하면서 반복자(Iterator) 를 사용한다.
순회 도중에 큐 구조를 변경하면 ConcurrentModificationException이나 예상치 못한 오류가 발생할 수 있다.

② 큐 상태가 바뀌면 순회가 이상해짐

큐에서 poll(), offer(), remove() 등으로 큐를 수정하면, 순회가 중간에 잘못될 수 있다.

위험한 예시: for-each에서 큐 수정

for (int[] doc : dq) {
    if (doc[1] > 3) {
        dq.poll(); // ❌ 큐를 수정하면서 순회
    }
}

큐 상태가 바뀌면서 큐 순회가 예상대로 진행되지 않는다.
큐 요소가 제거되면 Iterator의 상태가 일치하지 않아 오류가 발생할 수 있다.

안전한 방법: while 사용

while을 사용하면 큐의 상태를 변경하면서도 순회를 안정적으로 할 수 있다.

while (!dq.isEmpty()) {
    int[] doc = dq.poll();  // 큐 수정 가능
    dq.offer(doc);           // 큐 수정 가능
}

💻 전체 코드

// 큐에 [문서 인덱스, 중요도] 형태로 저장해야 함
// 1. 현재 문서보다 중요도가 더 높은 문서가 큐에 있으면 → 현재 문서를 뒤로 보냄
// 2. 그렇지 않으면 → 현재 문서를 인쇄하고 카운트를 증가
// 3. 인쇄한 문서가 내가 찾던 m번째 문서면 → 해당 카운트를 출력하고 종료

import java.io.*;
import java.util.*;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int t = Integer.parseInt(br.readLine());

        while (t-- > 0) { // for (int i = 0; i < t; i++) {와 동일 (t번 반복해라)
            StringTokenizer st = new StringTokenizer(br.readLine());
            int n = Integer.parseInt(st.nextToken());
            int m = Integer.parseInt(st.nextToken());

            st = new StringTokenizer(br.readLine());
            Deque<int[]> dq = new ArrayDeque<>();
            for (int i = 0; i < n; i++) {
                int priority = Integer.parseInt(st.nextToken());
                dq.offerLast(new int[]{i, priority});
            }

            // 출력된 문서 개수
            int printOrder = 0;

            // 큐가 빌 때까지 반복
            while (!dq.isEmpty()) {
                // 큐 맨 앞의 문서를 꺼냄
                int[] front = dq.pollFirst();

                // 이 문서보다 더 중요한 문서가 있는지 확인
                boolean hasMoreImportant = false;
                for (int[] doc : dq) {
                    if (doc[1] > front[1]) { // 중요도가 더 높은 문서 발견
                        hasMoreImportant = true;
                        break;
                    }
                }

                // 더 중요한 문서가 있으면 → 지금 문서를 뒤로 보냄
                if (hasMoreImportant) {
                    dq.offerLast(front); // 현재 문서를 큐 뒤에 다시 넣기
                } else {
                    // 더 중요한 문서가 없으면 → 이 문서를 인쇄
                    printOrder++; // 인쇄 순서 증가

                    // 방금 인쇄한 문서가 내가 찾던 m번째 문서면 출력 후 종료
                    if (front[0] == m) {
                        System.out.println(printOrder);
                        break;
                    }
                }
            }

        }
    }
}

💭 배운 점

처음에는 단순히 중요도만 고려하면 될 줄 알았는데,
문서의 위치를 함께 저장하지 않으면 내가 찾는 문서를 추적할 수 없다는 걸 알게 됐다.
그래서 [인덱스, 중요도] 형태로 배열을 저장해야 한다는 것을 알았다.
큐에 값을 그냥 넣는 게 아니라, 배열 형태로 [인덱스, 중요도]를 같이 저장할 수 있다는 걸 배웠다.
이렇게 하면 내가 찾는 문서가 어디 있는지 계속 추적할 수 있다.

🔁 복습 템플릿

Deque<int[]> dq = new ArrayDeque<>();
dq.offerLast(new int[]{index, priority}); // 위치 + 중요도 저장

int[] current = dq.pollFirst();

박시은