백준 1966 프린터큐 (JAVA 자바 풀이)

백준 1966 프린터큐 (JAVA 자바 풀이)

 

---

 

 

📌 문제 

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

 

 

⚔ 입력

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

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

 

 

📣 출력

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

 

 

 

 

---

 

 

 

💎 문제 분석

 

 

기존의 도큐먼트를 deque로 인덱스로 저장하고 같은 위치에 priority를 저장하는 deque를 따로 생성한다. 

 

이 두 가지를 같이 수정해줌으로써 hashmap에서 key와 value를 조정하는 것과 같은 효과를 내준다. 

 

 

괜찮은 반례 : 

1

3

1 1 1

답 : 2

 

1

5 1

1 2 3 1 2

답 : 3

 

1

5 0 

1 2 3 1 2

답 : 5

 

1

5 3 

1 2 3 1 2

답 : 4

 

 

 

💡 코드 구현

 

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.*;

public class Main {

    public static void main(String[] args) throws IOException {
        BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(System.in));

        int testCase = Integer.parseInt(bufferedReader.readLine());

        while (testCase-- > 0) {
            StringTokenizer stringTokenizer = new StringTokenizer(bufferedReader.readLine());
            int documents = Integer.parseInt(stringTokenizer.nextToken());
            int queryDocument = Integer.parseInt(stringTokenizer.nextToken());

            Deque<Integer> documentDeque = new ArrayDeque<>();
            Deque<Integer> priorityDeque = new ArrayDeque<>();

            stringTokenizer = new StringTokenizer(bufferedReader.readLine());

            for (int j = 0; j < documents; j++) {
                documentDeque.offer(j);
                priorityDeque.offer(Integer.valueOf(stringTokenizer.nextToken()));
            }

            printerSolution(queryDocument, documentDeque, priorityDeque);
        }
    }

    private static void printerSolution(int queryDocument, Deque<Integer> deque, Deque<Integer> priorityDeque) {
        int count = 1;

        while (!deque.isEmpty()) {
            int max = Collections.max(priorityDeque);
            int documentIndex = deque.pollFirst();
            int documentPriority = priorityDeque.pollFirst();

            if (documentPriority != max) {
                deque.offerLast(documentIndex);
                priorityDeque.offerLast(documentPriority);
            } else {
                if (documentIndex == queryDocument) {
                    System.out.println(count);
                    break;
                }
                count++;
            }
        }
    }
}