[Algorithm] 프로그래머스 - 5명씩 (Kotlin)

최대 5명씩 탑승 가능한 놀이기구를 타기 위해 줄을 서 있는 사람들의 이름이 담긴 문자열 리스트 names가 주어질 때, 앞에서부터 5명씩 묶은 그룹의 가장 앞에 서 있는 사람들의 이름을 담은 리스트를 리턴하는 함수를 완성하라. 마지막 그룹이 5명이 되지 않더라도 가장 앞 사람의 이름을 포함한다

 

 

주먹구구식으로 푼다면 아래와 같다.

 

class Solution {
    fun solution(names: Array<String>): ArrayList<String> {
        val result = names.toMutableList().chunked(5)
        val resultArr = arrayListOf<String>()
        repeat(result.size) {
            resultArr.add(result[it][0])
        }

        return resultArr
    }
}

 

코틀린의 chunked()를 사용하면 간단하게 풀 수 있다. 핵심은 사람 이름을 5개씩 묶은 리스트를 리턴하고 5명이 되지 않더라도 가장 앞 사람의 이름을 포함하는 리스트를 만드는 것인데, chunked()가 이 역할을 하는 함수다.

 

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

chunked - Kotlin Programming Language

이 컬렉션을 주어진 크기를 초과하지 않는 리스트의 리스트(List<List<T>>)로 분할한다. 결과 목록의 마지막 목록에는 지정된 크기보다 적은 수의 요소가 있을 수 있다

 

좀 더 간결하게 만들면 아래와 같다.

 

class Solution {
    fun solution(names: Array<String>): Array<String> {
        return names.toList().chunked(5) { it[0] }.toTypedArray()
    }
}