추억 점수

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

프로그래머스

[문제 설명]

사진들을 보며 추억에 젖어 있던 루는 사진별로 추억 점수를 매길려고 합니다. 사진 속에 나오는 인물의 그리움 점수를 모두 합산한 값이 해당 사진의 추억 점수가 됩니다. 예를 들어 사진 속 인물의 이름이 ["may", "kein", "kain"]이고 각 인물의 그리움 점수가 [5점, 10점, 1점]일 때 해당 사진의 추억 점수는 16(5 + 10 + 1)점이 됩니다. 다른 사진 속 인물의 이름이 ["kali", "mari", "don", "tony"]이고 ["kali", "mari", "don"]의 그리움 점수가 각각 [11점, 1점, 55점]]이고, "tony"는 그리움 점수가 없을 때, 이 사진의 추억 점수는 3명의 그리움 점수를 합한 67(11 + 1 + 55)점입니다.

그리워하는 사람의 이름을 담은 문자열 배열 name, 각 사람별 그리움 점수를 담은 정수 배열 yearning, 각 사진에 찍힌 인물의 이름을 담은 이차원 문자열 배열 photo가 매개변수로 주어질 때, 사진들의 추억 점수를 photo에 주어진 순서대로 배열에 담아 return하는 solution 함수를 완성해주세요.

 

제한사항

• 3 ≤ name의 길이 = yearning의 길이≤ 100
  • 3 ≤ name의 원소의 길이 ≤ 7
  • name의 원소들은 알파벳 소문자로만 이루어져 있습니다.
  • name에는 중복된 값이 들어가지 않습니다.
  • 1 ≤ yearning[i] ≤ 100
  • yearning[i]는 i번째 사람의 그리움 점수입니다.
• 3 ≤ photo의 길이 ≤ 100
  • 1 ≤ photo[i]의 길이 ≤ 100
  • 3 ≤ photo[i]의 원소(문자열)의 길이 ≤ 7
  • photo[i]의 원소들은 알파벳 소문자로만 이루어져 있습니다.
  • photo[i]의 원소들은 중복된 값이 들어가지 않습니다.

[나의 풀이]

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

 

[다른 풀이]

1)

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

 

2) 딕셔너리를 활용한 경우

def solution(name, yearning, photo):
    dictionary = dict(zip(name,yearning))
    scores = []
    for pt in photo:
        score = 0
        for p in pt:
            if p in dictionary:
                score += dictionary[p]
        scores.append(score)
    return scores

 

⭐dictionary = dict(zip(name, yearning))

• zip(name, yearning)은 name과 yearning 배열을 쌍으로 묶습니다.
• dict()를 사용하여 묶인 쌍을 딕셔너리로 변환합니다.

예시

name = ["may", "kein", "kain"]
yearning = [5, 10, 1]

zip(name, yearning) = [("may", 5), ("kein", 10), ("kain", 1)]

dictionary = {"may": 5, "kein": 10, "kain": 1}

 

3)

def solution(name, yearning, photo): 
    return [ sum([ { n: s for n, s in zip(name, yearning) }.get(a, 0) for a in p ]) for p in photo ]