[Kotlin/프로그래머스] 코딩테스트 연습 > 연습문제 > 명예의 전당 (1)

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
import java.util.PriorityQueue

class Solution {
fun solution(k: Int, score: IntArray): IntArray {
var answer = mutableListOf<Int>()
/**
* 내림차순 PriorityQueue를 만들고
* q.isEmpty or size < k면 s를 넣고 q[0]을 add
* 아니면
* q[0]<s면 q[0] add
* 아니면 s넣고 q[0] add
*/
val pq = PriorityQueue<Int>{o1, o2 -> o1.compareTo(o2)}

        score.forEach { s ->
            if (pq.isEmpty() || pq.size < k) {
                pq.offer(s)
                answer.add(pq.peek())
            } else {
                if (pq.peek() > s) {
                    answer.add(pq.peek())
                } else {
                    pq.poll()
                    pq.offer(s)
                    answer.add(pq.peek())
                }
            }
        }
        return answer.toIntArray()
    }
}