[Kotlin/프로그래머스] 코딩테스트 연습 > 스택/큐 > 프린터

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
import java.util.*

class Solution {
    fun solution(priorities: IntArray, location: Int): Int {
        /*
        큐에다가 Pair<index, prior> 형태로 넣고
        poll한거의 location이 맞으면 cnt 리턴
        max면 pop 하고 cnt ++ 아니면 다시 push (cnt 초기화는 1)
        max 체크는 descPriorities 리스트.desc로 sort 얘를 가리키는 인덱스도 필요
        */
        var answer = 1
        val queue: Queue<Pair<Int, Int>> = LinkedList()
        val descList = priorities.sortedDescending()
        var descIndex = 0
        priorities.forEachIndexed { index, priority ->
            queue.offer(priority to index)
        }
        while (queue.isNotEmpty()) {
            val item = queue.poll()
            if (item.second == location && item.first == descList[descIndex]) {
                return answer
            }
            if (item.first == descList[descIndex]) {
                answer ++
                descIndex ++
            } else {
                queue.offer(item)
            }
        }
        return answer
    }
}