[Programmers] Lv1.신고 결과 받기(kotlin)

문제

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

 

예시

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

 

 

 

풀이

• 해당 문제는 map, set 등의 자료구조를 잘 조합하면 쉽게 풀 수 있는 문제입니다.
• 먼저, id_report_set에 id에 하나의 set을 관리할 수 있는 map을 생성합니다.
• 이후 report의 하나씩 탐색하면서 id_report_set에 값을 세팅합니다. 중복된 사람은 자동으로 제거가 됩니다.
• reported_count에는 id 마다 신고받은 횟수를 기록합니다. 해당 값은 id_report_set을 탐색하여 id가 신고한 사람들을 담은 value 값으로 하나씩 횟수를 증가시킵니다.
• 마지막으로 id_list를 탐색하면서 id가 신고한 사람들 중 결과 메일을 받은 사람들의 수를 계산합니다.

  fun solution(id_list: Array<String>, report: Array<String>, k: Int): IntArray {
        var answer = IntArray(id_list.size)
        val id_report_set = mutableMapOf<String, Set<String>>().apply {
            id_list.forEach {
                put(it, setOf<String>())
            }
        }

        report.forEach {
            val str = it.split(" ")
            id_report_set.put(str[0], id_report_set.get(str[0])!! + setOf(str[1]))
        }

        val reported_count = mutableMapOf<String, Int>()
        for (elem in id_report_set) {
            val idset = elem.value
            idset.forEach {
                reported_count.put(it, (reported_count.get(it) ?: 0) + 1)
            }
        }

        id_list.forEachIndexed { index, elem ->
            val idset = id_report_set.get(id_list.get(index))!!
            answer[index] = idset.filter { reported_count.get(it)!! >= k }.size
        }

        return answer
    }

 

 

• 위처럼 풀 수도 있지만 kotlin의 다양한 함수들을 잘 알고 있다면 다음과 같이 메소드 체이닝만 사용하여 풀 수 있습니다!! 😎
• distinct, groupBy, flatten 모르는 메소드들이 많았기에 잘 기록해 둡시다~~ 

fun solution2(id_list: Array<String>, report: Array<String>, k: Int): IntArray =
            report.distinct().map { it.split(" ") }
                    .groupBy { it[1] }
                    .map { it.value }
                    .filter { it.size >= k }
                    .flatten()
                    .groupBy { it[0] }
                    .run { id_list.map { get(it)?.size ?: 0 }.toIntArray() }

 

 

 

참고

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

groupBy - Kotlin Programming Language

• https://kotlinlang.org/api/latest/jvm/stdlib/kotlin.collections/distinct.html

distinct - Kotlin Programming Language

• https://kotlinlang.org/api/latest/jvm/stdlib/kotlin.collections/flatten.html

flatten - Kotlin Programming Language