[프로그래머스] lv.1 추억 점수 python 풀이

문제

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

프로그래머스

 

문제 풀이

이름별 추억점수 dict를 만들어 photo 하위 배열마다 점수를 더함.

존재하지 않는 점수는 0으로 처리해서 sum함.

 

코드

def solution(name, yearning, photo):
    score_dict = dict()
    for i in range(len(name)):
        score_dict[name[i]] = yearning[i]
    
    op = lambda x: score_dict[x] if x in score_dict else 0
    res = []
    for case in photo: 
        res.append(sum([op(x) for x in case]))
    return res