[python] 프로그래머스 Lv1 - 추억 점수

🔗 문제 링크

프로그래머스

📂 분류

구현 자료구조

💡 풀이

이 문제는 map 자료구조를 알면 쉽게 풀 수 있다.

name과 yearning의 길이가 같기 떄문에 리스트를 순회하면서, 이름과 점수를 저장한다.
그리고 photo 이중 리스트를 순회하면서 dic에 있는 이름이라면 그 점수를 score에 더한다.

💻 코드

def solution(name, yearning, photo):
    answer = []
    dic = {}
    # 추억 점수 저장 O(N)
    for i in range(len(name)):
        dic[name[i]] = yearning[i]

    # 사진에 있는 사람들의 점수 계산 O(N^2) 10,000
    for row in photo:
        score = 0
        for n in row:
            if n in dic:
                score += dic[n]
        answer.append(score)

    return answer