알고리즘 문제 풀이/프로그래머스
[Algorithm] 프로그래머스 - 배열 원소의 길이 (Kotlin)
참깨빵위에참깨빵_
2022. 12. 30. 16:22
728x90
반응형
매개변수로 주어진 문자열 배열 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()
}
반응형