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

SW Test/Programmers

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

An effort will never betray 😎 2022. 7. 23. 00:35

문제

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() }