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

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

제한사항

• 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]의 원소들은 중복된 값이 들어가지 않음

 

"""
입출력 예시)

name = ["may", "kein", "kain", "radi"], yearning = [5, 10, 1, 3], photo = [["may", "kein", "kain", "radi"],["may", "kein", "brin", "deny"], ["kon", "kain", "may", "coni"]] -> [19, 15, 6]
name = ["kali", "mari", "don"], yearning = [11, 1, 55], photo = [["kali", "mari", "don"], ["pony", "tom", "teddy"], ["con", "mona", "don"]] -> [67, 0, 55]
name = ["may", "kein", "kain", "radi"], yearning = [5, 10, 1, 3], photo = [["may"],["kein", "deny", "may"], ["kon", "coni"]] -> [5, 15, 0]
"""

 

 

## 의사코드 ##

# 인물별 그리움 점수를 딕셔너리로 저장
#  dic = {k:v for k, v in zip(name, yearning)}

# 사진 속 인물의 이름이 그리움 점수가 없을 때는 통과
# if p not in dic.keys():
#     continue

 

 

 

통과한 코드

 

def solution(name, yearning, photo):
    answer = []
    # 인물별 그리움 점수를 담은 딕셔너리
    dic = {k:v for k, v in zip(name, yearning)}
    for ph in photo:
        cnt = 0 # 사진의 추억 점수
        for p in ph:
            # 사진 속 인물의 이름이 그리움 점수가 없을 때는 통과
            if p not in dic.keys():
                continue
            else:
                cnt += dic[p]
        answer.append(cnt)
    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]

 

 

-> index()와 sum()을 활용해 한 줄로 작성