1966 프린터 큐 (Java)

🌍***TISTORY***

문제 🌐

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

입력

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

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

출력

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

예시 입력

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

예시 출력

1
2
5

접근 방법

큐를 두 개 만들어서 인덱스와 우선순위를 함께 가져갔다.

			for (int fast = 9; fast > 0; fast--) {
				boolean isComplete = false;
				while (q.contains(fast)) {
					int check = q.poll();
					int cIdx = idx.poll();
					if (check == fast && cIdx == M) {
						System.out.println(order);
						isComplete = true;
						break;
					} else if (check == fast) {
						order++;
					} else {
						q.offer(check);
						idx.offer(cIdx);
					}
				}
				if (isComplete)
					break;
			}

우선순위는 1부터 9까지 정수로 주어진다고 했기 때문에 우선순위 역순으로 반복문을 돌면서 확인했다. 가장 높은 우선순위가 있는지 확인하고, 있다면 그 우선순위에 해당하는 문서가 나올 때까지 큐에서 꺼내서 확인했다.

그리고 확인하고 싶은 인덱스가 맞고, 현재 확인 중인 우선순위가 맞다면 원하는 인덱스의 순서를 찾았으므로 isComplete을 true로 바꾸고 반복문을 break한다.

현재 확인 중인 우선순위를 갖는 문서가 나온다면 큐에서 꺼내고 순서에 1을 더해줬다.

두 경우에도 해당되지 않는다면, 확인하고 싶은 인덱스의 차례가 오지 않았거나 우선순위에서 후순위이므로 큐의 가장 뒤로 다시 넣어줬다.

순서를 나타내는 변수 order를 초기화할 때 무의식적으로 0으로 정해 줬는데 0번째 순서는 없으므로… 1로 초기화해 줬어야 했다.

Full Solution

import java.util.Scanner;
import java.util.LinkedList;
import java.util.Queue;

public class Main {
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int t = sc.nextInt();
		while (t-- > 0) {
			int N = sc.nextInt();
			int M = sc.nextInt();
			Queue<Integer> q = new LinkedList<>();
			Queue<Integer> idx = new LinkedList<>();

			for (int i = 0; i < N; i++) {
				q.offer(sc.nextInt());
				idx.offer(i);
			}
			int order = 1;
			for (int fast = 9; fast > 0; fast--) {
				boolean isComplete = false;
				while (q.contains(fast)) {
					int check = q.poll();
					int cIdx = idx.poll();
					if (check == fast && cIdx == M) {
						System.out.println(order);
						isComplete = true;
						break;
					} else if (check == fast) {
						order++;
					} else {
						q.offer(check);
						idx.offer(cIdx);
					}
				}
				if (isComplete)
					break;
			}
		}
	}
}