POG 추억 점수

링크

코드

class Solution {
    fun solution(name: Array<String>, yearning: IntArray, photo: Array<Array<String>>): IntArray {
        val answer = ArrayList<Int>()

        val memoryMap = name.mapIndexed { index, s -> s to yearning[index] }.toMap()

        photo.forEach photo@ {people ->
            var memorySum: Int = 0
            people.forEach people@ { person ->
                memorySum += memoryMap[person] ?: 0
            }
            answer.add(memorySum)
            memorySum = 0
        }

        return answer.toIntArray()
    }
}

class Solution1 {
    fun solution(names: Array<String>, yearnings: IntArray, photoes: Array<Array<String>>): IntArray {
        val map = names.zip(yearnings.toTypedArray()).toMap()
        return photoes.map { photo -> photo.sumOf { map[it] ?: 0 } }.toIntArray()
    }
}

Solution1 은 압축버전. 다른사람이 짠건데 테크닉이 좋아보여서 가져옴