프로그래머스 Lv.1 - 추억 점수

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

프로그래머스

파이썬의 dictionary 자료형 문법을 잘 활용할 줄 알면 간단하게 풀 수 있는 문제이다.

dictionary에서 key를 가지고 값을 가져올 때, dict[key] 로 가져오는 경우 해당 Key가 없으면 에러를 발생시킨다. 에러를 발생시키고 싶지 않은 경우에는 dict.get(key) 로 가져오면 None을 반환한다. 

 

def solution(name, yearning, photo):
    yearning_dict = {}
    answer = []
    for i in range(len(name)):
        yearning_dict[name[i]] = yearning[i]
    for i in range(len(photo)):
        score = 0
        for j in range(len(photo[i])):
            if (yearning_dict.get(photo[i][j])): score += yearning_dict[photo[i][j]]
        answer.append(score)
    
    return answer