[level 1] 추억 점수

문제

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

 

 

문제 풀이 접근법1

1) name과 yearning 리스트를 합쳐서 key는 name이, value에는 yearning이 들어가도록 하는 딕셔너리를 만든다.

2) 열, 행의 길이 만큼 반복하는 이중 반복문으로 이차원 리스트 photo의 원소에 접근한다.

3) sum = 0을 두번째 반복문을 시작하기 전에 선언한다.

4) photo의 각 원소가 score 딕셔너리에 존재하는지 확인한다. 만약 존재한다면 해당 값을 sum에 더하고, 없다면 0을 더한다.

5) 각 행의 추억점수를 answer에 추가한다. 

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

    for i in range(len(photo)):
        sum = 0
        for j in range(len(photo[i])):

            if photo[i][j] in score:
                sum += score[photo[i][j]]
        
            else:
                sum += 0
    
        answer.append(sum)
        
    return answer

 

---

 

다른 분들 풀이

def solution(이름, 점수, 사진):
    return [sum(점수[이름.index(j)] for j in i if j in 이름) for i in 사진]