[프로그래머스 Lv1] 추억 점수(python, 해시)

[문제]

[풀이]

def solution(name, yearning, photo):
    answer = []
    for i in range(len(photo)):
        missing_score = 0
        for j in photo[i]:
            if j in name:
                missing_score += yearning[name.index(j)]
        answer.append(missing_score)
    return answer

나는 인덱싱으로 풀었지만 시간복잡도가 O(P⋅N^2) 이므로 시간초과날 확률 높음! 

아래는 해시로 다시 풀어본 것, 아래코드는 시간복잡도 O(N + P⋅M)

def solution(name, yearning, photo):
    score_map = dict(zip(name, yearning))
    
    answer = []
    for people in photo:
        total = sum(score_map.get(person, 0) for person in people)
        answer.append(total)
    
    return answer

[링크]

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

프로그래머스