추억 점수

• 문제 설명

사진들을 보며 추억에 젖어 있던 루는 사진별로 추억 점수를 매길려고 합니다. 사진 속에 나오는 인물의 그리움 점수를 모두 합산한 값이 해당 사진의 추억 점수가 됩니다. 예를 들어 사진 속 인물의 이름이 ["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 함수를 완성해주세요.

def solution(name, yearning, photo):
    na = name
    year = yearning
    temp = []
    for i in range(3):
        answer = []
        for j in photo[i]:
            count = 0
            if j in na:
                idx = na.index(j)
                count += year[idx]
                answer.append(count)
        temp.append(sum(answer))
    print(temp)
    return temp

 

테스트는 진행되지만 실패.

def solution(name, yearning, photo):
    # 1. 반환할 결괏값 선언과 이름과 그리움 점수를 묶어준다.
    result = []
    info = dict(zip(name, yearning))
    # 2. 사진 속 사람들을 확인할 for loop
    for people in photo:     
        # 3. 그리움 점수 누적을 위한 변수 선언
        score = 0      
        # 4. 사람들 각각의 점수를 확인할 for loop
        for person in people:
            # 5. 사람 별 점수를 누적, 없는 사람은 0
            score += info.get(person, 0)
        # 6. 결과 배열에 담아주기
        result.append(score)
    
    return result