[Programmers] Lv1. 추억점수(kotlin)

문제

 

 

예시

 

 

풀이

• name에 있으면 해당 점수를 더하고 없으면 0으로 더하면 답을 구할 수 있습니다.

class Solution {
    fun solution(name: Array<String>, yearning: IntArray, photo: Array<Array<String>>): IntArray =
        photo.map{ names -> 
            names.foldIndexed(0){ index, acc, i -> 
                acc + if(name.indexOf(i) >= 0) yearning[name.indexOf(i)] else 0
            }
        }.toIntArray()
}

 

 

참고

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

foldIndexed - Kotlin Programming Language