[Programmers] Lv1. 2019 KAKAO BLIND RECUITMENT 실패율(kotlin)

문제

https://programmers.co.kr/learn/courses/30/lessons/42889?language=kotlin

 

예시

https://programmers.co.kr/learn/courses/30/lessons/42889?language=kotlin

 

풀이

1. 각 스테이지별로 머무는 사용자의 수를 배열로 표현 ( N+1 까지 있을 수 있어서 N+2 크기의 배열 생성 )
2. 해당 스테이지에 머무는 사용자 수 / 스테이지를 도전하는 총 사용자의 수를 계산
3. 스테이지의 실패율이 높으면서 스테이지 번호가 낮은 순으로 정렬 후 스테이지 번호만 출력

class Solution {
    fun solution(N: Int, stages: IntArray): IntArray {
        val answer = IntArray(N+2)	// 스테이지에 머무는 사용자를 담는 배열
        stages.forEach{ answer[it]++ }	
        
        var total = stages.size
        val list = mutableListOf<Pair<Int, Double>>()
        for(i in 1..N){		// 스테이지에 머무는 사용자 수 / 스테이지 도전하는 총 사용자 수
            list.add(Pair(i, answer.get(i) / total.toDouble()))
            total -= answer.get(i)
        }
       	
        // 스테이지 실패율이 높으면서 스테이지 번호가 낮은 순으로 정렬
        return list.sortedWith(compareBy({-it.second}, {it.first})).map { it.first }.toIntArray()
    }
}

 

참고

• 함수형 프로그래밍 언어인 kotlin의 특성에 맞게 함수를 많이 익히자 😽
• https://kotlinlang.org/api/latest/jvm/stdlib/kotlin.collections/sorted-with.html

sortedWith - Kotlin Programming Language

• https://kotlinlang.org/api/latest/jvm/stdlib/kotlin.comparisons/compare-by.htm

compareBy - Kotlin Programming Language