[백준/BOJ] 1966번: 프린터 큐

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

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

해설

문제를 봐보면

큐에 프린트 할 문서들이 있을 때 중요도가 높은 것부터 뽑아야한다.

맨 앞의 문서보다 중요도가 높은 문서가 뒤에 있을 시 맨 앞의 문서를 맨 뒤로 보내고 반복하며 중요도가 가장 높은 문서가 맨 앞으로 오게해야 한다.

나는 초기 큐에 있던 값들과 인덱스를 한번에 저장하기 위해 que: LinkedList<Pair<Int, Int>>를 사용하였다.

우선 que에 문서들의 인덱스와 중요도를 저장하자.

다음에 첫번째 문서를 poll()하여 이 문서와 나머지 문서들 중 더 큰 중요도가 나올 때까지 반복한다.

더 큰 중요도를 만나면 poll()한 문서와 큰 중요도를 갖는 문서의 앞에 있는 문서들을 모두 뒤로 보낸다.

다시 poll()을 하여 이전 작업을 반복하여 중요도에 맞게 배치가 되었다면 이제 인덱스의 값이 맞는 곳을 찾아주면 된다.

글로 보기에 이해가 쉽지 않을 수 있다. 코드와 같이 이해를 하면 좀 더 수월할 것이다.

소스 코드

import java.util.*

fun main() {
    val testCase = readLine()!!.toInt()
    for(tc in 0 until testCase) {
        val st = StringTokenizer(readLine())
        val n = st.nextToken().toInt()
        val m = st.nextToken().toInt()
        val que: LinkedList<Pair<Int, Int>> = LinkedList()
        val input = readLine()!!.split(" ")
        for(i in input.indices) que.add(Pair(i, input[i].toInt()))
        var count = 0
        while(!que.isEmpty()) {
            val front: Pair<Int, Int> = que.poll()
            var isMax = true
            for(i in 0 until que.size) {
                if(front.second < que[i].second) {
                    que.add(front)
                    for(j in 0 until i) que.add(que.poll())
                    isMax = false
                    break
                }
            }
            if(!isMax) continue
            count++
            if(front.first == m) break
        }
        println(count)
    }
}