백준 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번째로 인쇄되게 된다.

---

코드

import java.io.*;
import java.util.LinkedList;
import java.util.Queue;
import java.util.StringTokenizer;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br=new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw=new BufferedWriter(new OutputStreamWriter(System.out));
        Queue queue=new LinkedList();

        int n=Integer.parseInt(br.readLine());
        for(int i=0; i<n; i++){
            int count=0;
            String testCase=br.readLine();
            StringTokenizer st=new StringTokenizer(testCase);
            int num=Integer.parseInt(st.nextToken());
            int goal=Integer.parseInt(st.nextToken());
            testCase=br.readLine();
            st=new StringTokenizer(testCase);
            int[][] weight=new int[num][2];
            for(int j=0; j<num; j++) {
                weight[j][1] = Integer.parseInt(st.nextToken());
                weight[j][0]=j;
                queue.add(j);
            }

            while(true){
                int max=-1;
                for(int j=0; j<num; j++){
                    int temp= (int) queue.remove();
                    if(max==-1){
                        max=temp;
                        queue.add(temp);
                        continue;
                    }
                    if(weight[max][1]<weight[temp][1])
                        max=temp;
                    queue.add(temp);
                }
                if(max==goal){
                    count++;
                    break;
                }

                while(true){
                    int temp= (int) queue.remove();
                    if(temp==max) {
                        count++;
                        break;
                    }
                    else
                        queue.add(temp);
                }
            }

            bw.write(count+"\n");
            queue.clear();
        }
        bw.flush();
    }
}

---

해결 과정

1. 처음에는 단순히 중요도가 높은 순서로 몇 번째에 해당하는지 찾으면 되는 문제인 줄 알았지만, 같은 중요도의 요소가 있을 경우 LIFO를 따라서 몇 번째에 해당하는 지가 핵심이라는 것을 알았다. 그래서 Queue를 사용해서 문제의 조건대로 진행하면서 원하는 요소가 출력될 때를 찾으려고 했는데 Queue에서 하나씩 출력될 때마다 원하는 요소의 인덱스가 처음의 인덱스와 달라져서 내가 원하는 요소인지, 단순히 중요도만 같은 요소인지 판단할 수 없었다. 따라서 중요도 배열을 2차원으로 선언하여 처음의 순서를 저장해두고 중요도와 함께 비교하였다.
   
   구현은 현재 큐에 있는 요소들 중 가장 큰 요소를 찾고 remove를 하면서 문제에서 요구하는 대로 동작하도록 했다. 그러다가 가장 큰 요소가 출력하려는 요소와 같으면 test case를 종료하도록 했다.
2. 😁