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()
    }
}
반응형