230723 프로그래머스 Lv1 - 추억 점수

프로그래머스

문제

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

2. photo 에서 각 사진을 돌며 추억 점수가 있는 인물이 있는지 확인하고,

3. 있다면 점수를 더한다.

해결

def solution(name, yearning, photo):
    answer = []
    yearning_dict = {name[i]:yearning[i] for i in range(len(name))}
    for list in photo: # 사진 하나씩
        score = 0
        for key in list: # 사진 속 인물 하나씩
            if key in yearning_dict:
                score+= yearning_dict[key]
        answer.append(score)
    return answer

0801 추가 - 보완

for문과 if 문이 반복되는 것이 보기 싫어서 풀이를 보완해보기로 했다.

불필요한 for 문을 줄이고, 대신 내장 함수인 zip, sum, get을 사용하여 작업을 간략화함

1. yearning_dict를 생성할 때 딕셔너리 컴프리헨션 대신 zip 함수를 사용
2. 리스트 컴프리헨션을 사용하여 answer 리스트를 생성

 - 각 photo 리스트에 대해 사진 속 인물 하나씩 순회하며 yearning_dict에서 해당 키(key)에 해당하는 값을 가져와서 모두 더한다.

def solution(name, yearning, photo):
    yearning_dict = dict(zip(name, yearning))
    answer = [sum(yearning_dict.get(key, 0) for key in lst) for lst in photo]
    return answer