[백준1966번] 프린터 큐

문제

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

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

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

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

입력

첫 줄에 test case의 수가 주어진다. 각 test case에 대해서 문서의 수 N(100이하)와 몇 번째로 인쇄되었는지 궁금한 문서가 현재 Queue의 어떤 위치에 있는지를 알려주는 M(0이상 N미만)이 주어진다. 다음줄에 N개 문서의 중요도가 주어지는데, 중요도는 1 이상 9 이하이다. 중요도가 같은 문서가 여러 개 있을 수도 있다. 위의 예는 N=4, M=0(A문서가 궁금하다면), 중요도는 2 1 4 3이 된다.

출력

각 test case에 대해 문서가 몇 번째로 인쇄되는지 출력한다.

예제 입력 1

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

예제 출력 1

1
2
5

나의 풀이

필자는 이 문제를 Paper 라는 class를 만들어 문서의 순서로 매긴 num과 중요도를 pri로 각각 저장을 해준 뒤 이 Paper들을 queue에 넣었다. 그리고 중요도를 따로 배열에 저장하여 내림차순 정렬을 하였고 이를 통하여 높은 중요도와 queue에서 앞에 있는 Paper들을 먼저 삭제해 나갔다.

중요도를 내림차순으로 정렬해 놓았기 때문에 높은 중요도부터 나오게되고 중요도와 Paper의 중요도가 동일할 경우 다시 queue의 뒷 부분에 넣지않는다 또한 다르다면 queue의 뒷 부분에 넣는다. 이렇게 한 번 돌때마다 count를 1씩증가시키며 Paper의 중요도와 배열에서 꺼내진 중요도가 같고 M이 Paper의 num과 같은 경우 그때의 count에 1을 더한값을 출력시켜주면 된다.

코드

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

/**
 * 1966번 프린터 큐
 */

public class Baekjoon1966 {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = null;

        int t = Integer.parseInt(br.readLine());

        for (int i=0; i<t; i++) {
            st = new StringTokenizer(br.readLine());
            int n = Integer.parseInt(st.nextToken());
            int m = Integer.parseInt(st.nextToken());

            List<Paper> queue = new ArrayList<>();
            Integer[] priority = new Integer[n];

            st = new StringTokenizer(br.readLine());
            for (int j=0; j<n; j++) {
                int tmp_pri = Integer.parseInt(st.nextToken());
                queue.add(new Paper(j,tmp_pri));
                priority[j] = tmp_pri;
            }

            Arrays.sort(priority, Collections.reverseOrder());

            int count = 0;
            while (queue.size() != 0) {
                Paper front = queue.remove(0);
                if (priority[count] == front.pri) {
                    if (front.num == m) {
                        System.out.println(count+1);
                        break;
                    }
                    count++;
                } else {
                    queue.add(front);
                }
            }
        }
    }
}

class Paper{
    int num, pri;

    public Paper(int num, int pri) {
        this.num = num;
        this.pri = pri;
    }
}