[Algorithm] 프로그래머스 - 무작위로 K개의 수 뽑기 (Kotlin)

 

풀지 못해서 다른 사람의 풀이를 참고했다.

 

class Solution {
    fun solution(arr: IntArray, k: Int) = arr.distinct().toMutableList().apply {
        if (size > k) return subList(0, k)
        else repeat(k - size) { add(-1) }
    }
}

class Solution {
    fun solution(arr: IntArray, k: Int): IntArray {
        val result = mutableListOf<Int>()
        val seen = mutableSetOf<Int>()

        for (num in arr) {
            if (num !in seen) {
                seen.add(num)
                result.add(num)
            }
            if (result.size == k) {
                break
            }
        }

        while (result.size < k) {
            result.add(-1)
        }

        return result.toIntArray()
    }
}