[프로그래머스] 추억 점수 - Python

1. 문제

https://school.programmers.co.kr/learn/courses/30/lessons/176963

프로그래머스

사진들을 보며 추억에 젖어 있던 루는 사진별로 추억 점수를 매길려고 합니다. 사진 속에 나오는 인물의 그리움 점수를 모두 합산한 값이 해당 사진의 추억 점수가 됩니다. 예를 들어 사진 속 인물의 이름이 ["may", "kein", "kain"]이고 각 인물의 그리움 점수가 [5점, 10점, 1점]일 때 해당 사진의 추억 점수는 16(5 + 10 + 1)점이 됩니다. 다른 사진 속 인물의 이름이 ["kali", "mari", "don", "tony"]이고 ["kali", "mari", "don"]의 그리움 점수가 각각 [11점, 1점, 55점]]이고, "tony"는 그리움 점수가 없을 때, 이 사진의 추억 점수는 3명의 그리움 점수를 합한 67(11 + 1 + 55)점입니다.

그리워하는 사람의 이름은 담은 문자열 배열 name, 각 사람별 그리움 점수를 담은 정수 배열 yearning, 각 사진에 찍힌 인물의 이름을 담은 이차원 문자열 배열 photo가 매개변수로 주어질 때, 사진들의 추억 점수를 photo에 주어진 순서대로 배열에 담아 return하는 solution  함수를 완성해주세요.

 

2. Python code

def solution(name, yearning, photo):
    answer = []

    for ppl in photo:
        score = 0
        for n in ppl:
            if n in name:
                score += yearning[name.index(n)]
        answer.append(score)

    return answer

 

3. 풀이

name.index(n)을 해주면 해당 이름의 인덱스 값이 나온다.

위의 이름으로 예를 들면 may의 name.index(n)은 0, kein은 1로 이름의 인덱스 값이 뽑힌다.

이렇게 뽑은 인덱스 값으로 yearning[0] 해주면 5, yearning[1] 하면 10으로 각 이름의 추억점수가 나온다.