코딩테스트

[Python] 추억 점수

snoony 2024. 3. 5. 12:16

def solution(name, yearning, photo):
    answer = []
    name_dict = dict(zip(name, yearning))
    for each_photo in photo:
        sum = 0
        for each_name in each_photo:
            if each_name in name_dict:
                sum += name_dict[each_name]
        answer.append(sum)
    return answer

dict(zip(name, yearning)) : name, yearning 배열을 하나의 dictionary로 만들기

zip : 두개의 리스트를 묶어줄 때 사용

해당 키 값이 딕셔너리에 있는지 확인 : if each_name in name_dict