코딩 테스트/Level 1
프로그래머스 / 추억 점수
컴닥
2023. 4. 1. 07:47
반응형
https://school.programmers.co.kr/learn/courses/30/lessons/176963
파이썬
def solution(name, yearning, photo):
scores = {n: s for n, s in zip(name, yearning)}
return [sum(scores.get(person, 0) for person in pic) for pic in photo]
def solution(name, yearning, photo):
return [sum({n: s for n, s in zip(name, yearning)}.get(person, 0) for person in pic) for pic in photo]
코틀린
class Solution {
fun solution(name: Array<String>, yearning: IntArray, photo: Array<Array<String>>): IntArray {
return photo.map {
it.sumOf { each ->
name.toList()
.zip(yearning.toList())
.toMap()
.getOrDefault(each, 0) }
}.toIntArray()
}
}
반응형