[백준_1966] 프린터 큐 - JAVA

문제 출처

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

---

풀이

• 우선순위 큐와 큐를 사용하여 풀 수 있는 문제이다.
• 큐에 index와 우선순위를 넣는다. 그리고 우선순위 큐에는 우선순위의 값만 넣는다.
• 큐를 poll() 하면서 우선순위 큐의 peek() 값과 비교하여 일치하는지를 확인한다.
• 만약 일치한다면 큐에서 뽑은 데이터의 index가 M과 맞는지를 비교 후 값을 출력한다. 그렇지 않다면 다시 큐에 add() 한다.

---

코드

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

public class Main {
    private static class Data {
        int number;
        int order;

        public Data(int number, int order) {
            this.number = number;
            this.order = order;
        }
    }

    public static void main(String[] args) throws IOException {
        BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer stringTokenizer = new StringTokenizer(bufferedReader.readLine());
        int T = Integer.parseInt(stringTokenizer.nextToken());
        while (T-- > 0) {
            stringTokenizer = new StringTokenizer(bufferedReader.readLine());
            int N = Integer.parseInt(stringTokenizer.nextToken());
            int M = Integer.parseInt(stringTokenizer.nextToken());
            Queue<Data> queue = new LinkedList<>();
            PriorityQueue<Integer> priorityQueue = new PriorityQueue<>((o1, o2) -> o2 - o1);
            stringTokenizer = new StringTokenizer(bufferedReader.readLine());
            for (int i = 0; i < N; i++) {
                int value = Integer.parseInt(stringTokenizer.nextToken());
                queue.add(new Data(i, value));
                priorityQueue.add(value);
            }
            int count = 0;
            while (!queue.isEmpty()) {
                Data data = queue.poll();
                if (priorityQueue.peek() == data.order) {
                    priorityQueue.poll();
                    count++;
                    if (data.number == M) {
                        System.out.println(count);
                    }
                } else {
                    queue.add(data);
                }
            }
        }
    }
}

---

채점 결과