Algorithm/프로그래머스
[프로그래머스>LV1] 추억점수(Kotlin)
RIEN😚
2023. 3. 31. 09:27
728x90
반응형
프로그래머스
코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.
programmers.co.kr
1. 문제
[입력]
- 그리워하는 사람의 이름이 담은 문자열 배열 name
- 각 사람별 그리움 점수를 담은 정수 배열 yearning
- 각 사진에 찍힌 인물의 이름을 담은 이차원 문자열 배열 photo
[결과]
사진들의 추억 점수를 photo에 주어진 순서대로 배열에 담아 return
2. 코드
Kotlin
class Solution {
fun solution(
name: Array<String>,
yearning: IntArray,
photo: Array<Array<String>>
): IntArray {
val n = name.size
val people = hashMapOf<String, Int>()
(0 until n).forEach {
people[name[it]] = yearning[it]
}
return photo
.map { it.sumOf { person -> people[person] ?: 0 } }
.toIntArray()
}
}
반응형