[프로그래머스] 추억 점수 Python 풀이

---

문제

---

구현

dict에 사람과 그리움 점수를 담고

photo에서 계산하면서 result에 담으면 된다.

사실 레벨 0으로 들어가도 될 것 같은 문제다.

 

---

코드

from collections import defaultdict

def solution(name, yearning, photo):
    ndict = defaultdict(int)
    answer = []
    for n, y in zip(name, yearning):
        ndict[n] = y
    
    for p in photo:
        sumyearn = 0
        for n in p:
            sumyearn += ndict[n]
        answer.append(sumyearn)
            
    return answer

---

다른 풀이

파이썬의 또다른 재미는 숏코딩이다.

def solution(name, yearning, photo):
    return [sum(yearning[name.index(j)] for j in i if j in name) for i in photo]

싫어하는 분들도 많으니 따로 좋다고 말은 못 하지만 재미있다.