[백준 알고리즘] 11726번 : 프린터 큐

문제

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

출력

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

코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.LinkedList;
import java.util.StringTokenizer;

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

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

        for (int c = 0; c < tc; c++) {
            st = new StringTokenizer(br.readLine(), " ");
            int N = Integer.parseInt(st.nextToken());       //문서 개수
            int M = Integer.parseInt(st.nextToken());       //찾고자 하는 문서 인덱스

            LinkedList<int[]> queue = new LinkedList<>();

            st = new StringTokenizer(br.readLine(), " ");
            for (int i = 0; i < N; i++) {
                //{초기위치(인덱스), 중요도}. 중요도를 큐에 저장.
                queue.offer(new int[]{i, Integer.parseInt(st.nextToken())});
            }

            int count = 0;
            while (!queue.isEmpty()) {
                int[] front = queue.poll();
                boolean isMax = true;

                for (int i = 0; i < queue.size(); i++) {
                    if (front[1] < queue.get(i)[1]) {
                        queue.offer(front);
                        for (int j = 0; j < i; j++)
                            queue.offer(queue.poll());
                        isMax = false;
                        break;
                    }
                }

                if (isMax == false) continue;

                ++count;

                if (front[0] == M) break;
            }
            System.out.println(count);
        }
    }
}

풀이 및 느낀점

LinkedList를 이용한 queue를 생성했다. queue에 각 문서의 중요도를 저장했는데, 1차원 배열을 이용하여 초기위치와 중요도를 함께 저장했다.
큐가 비어있지않는 동안 반복문을 돈다. 먼저 queue의 가장 앞 부분을 빼내서 front 배열에 저장한다. queue의 크기만큼 반복문을 돌면서 queue의 전체 값들과 front에 저장한 값을 비교한다.
만약 front 값의 중요도가 queue(i)의 중요도보다 작다면, 큐에서 제거했던 front 값을 다시 queue에 저장한다. 이는 front보다 중요도가 더 큰 문서가 있다는 뜻이므로, front의 출력을 뒤로 미룬다는 것이다. 그리고 isMax값을 false로 수정한다. front가 최대값이 아니었다는 뜻이다.
front가 최대값이 맞다면, 바로 출력을 해주면 되기 때문에 count를 증가시킨다. 또한 front의 위치와 찾고자하는 문서의 위치가 같다면, count를 출력해준다.

참고자료