[프로그래머스 lv1] 추억 점수

문제 보기

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

프로그래머스

 

문제 해결

dic = {}
for i in range(len(name)):
    dic[name[i]] = yearning[i]

위의 소스 코드를 아래 한 줄로 작성할 수 있다.

dic = dict(zip(name, yearning))

 

소스 코드

def solution(name, yearning, photo):
    answer = []
    dic = dict(zip(name, yearning))
        
    for p in photo:
        sum = 0
        for i in range(len(p)):
            if p[i] in dic:
                sum += dic[p[i]]
        answer.append(sum)    
            
    return answer