[Algorithm] 프로그래머스 - 배열 원소의 길이 (Kotlin)

매개변수로 주어진 문자열 배열 strlist 각 원소의 길이를 담은 배열을 리턴하는 solution()을 완성하라

 

 

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

 

class Solution {
    fun solution(strList: Array<String>): IntArray {
        val a = arrayListOf<Int>()
        for (i in strList) {
            a.add(i.length)
        }

        return a.toIntArray()
    }
}

 

이것을 간결하게 만든다면 아래처럼 할 수 있다.

 

class Solution {
    fun solution(strList: Array<String>) = strList.map { it.length }.toIntArray()
}