문제 설명
사진들을 보며 추억에 젖어 있던 루는 사진별로 추억 점수를 매길려고 합니다. 사진 속에 나오는 인물의 그리움 점수를 모두 합산한 값이 해당 사진의 추억 점수가 됩니다. 예를 들어 사진 속 인물의 이름이 ["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
["may", "kein", "kain", "radi"]
yearning
[5, 10, 1, 3]
photo
[["may", "kein", "kain", "radi"],
["may", "kein", "brin", "deny"],
["kon", "kain", "may", "coni"]]
result
["may", "kein", "brin", "deny"]
그리워하는 사람의 이름을 담은 문자열 배열 name, 각 사람별 그리움 점수를 담은 정수 배열 yearning, 각 사진에 찍힌 인물의 이름을 담은 이차원 문자열 배열 photo가 매개변수로 주어질 때, 사진들의 추억 점수를 photo에 주어진 순서대로 배열에 담아 return하는 solution 함수를 완성해주세요.
문제 풀이
name: 그리워하는 사람 배열(여기없으면 안그리워함)
yearning: name배열에 인덱스에 맞춰서 순서대로 그리워하는 정도
photo: 사진속 사람들(name 배열에 없는 사람은 그리워하는 정도 0임)
방법 => 딕셔너리 이용
1. 일단 두 리스트 name과 yearning의 원소들을 각각 key,value으로 처리해서 하나의 딕셔너리로 만들고
2. photo 돌면서(이차원 배열->이중for문 필요) 딕셔너리에 각 이름(key)들이 있는지 검사
3. 검사했을때 딕셔너리에 존재하면 대응되는 그리움 정도(value)를 누적합에 더함
4. 단일 배열 하나 돌때마다 마지막에 result배열에 누적합을 원소로추가(추가하고 초기화 필요)
def solution(name, yearning, photo):
answer=[]
#딕셔너리 생성
name_arr = dict(zip(name, yearning))
for arr in photo:
sum = 0
for name in arr:
if(name in name_arr): sum += name_arr[name]
answer.append(sum)
return answer