본문 바로가기
코딩테스트/python

프로그래머스 lv1 추억 점수

by 앵유짱 2023. 8. 25.

문제 설명

사진들을 보며 추억에 젖어 있던 루는 사진별로 추억 점수를 매길려고 합니다. 사진 속에 나오는 인물의 그리움 점수를 모두 합산한 값이 해당 사진의 추억 점수가 됩니다. 예를 들어 사진 속 인물의 이름이 ["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 함수를 완성해주세요.

 

입출력 예

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

 

 

def solution(name, yearning, photo):
	
    # zip을 이용해서 두 list를 dictionary로 묶어줌
    name_score = dict(zip(name,yearning))
    result = []
    
    # photo는 2차원 배열이므로 한번씩 for문을 돌아줌
    for people in photo :
    
    	# 점수값 넣을 곳 초기화
        photo_sum = 0
        
        # people에서도 하나하나 보기위해 한번더 for문
        for name in people :
        
        	# 돌면서 name이 name_socre에 있는 녀석이면 해당 name_score[name], 즉 value를 photo_sum에 누적
            if name in name_score :
                photo_sum = photo_sum + name_score[name] 
                
        result.append(photo_sum)
    return result