본문 바로가기

코딩테스트

[Python] 추억 점수

def solution(name, yearning, photo):
    answer = []
    name_dict = dict(zip(name, yearning))
    for each_photo in photo:
        sum = 0
        for each_name in each_photo:
            if each_name in name_dict:
                sum += name_dict[each_name]
        answer.append(sum)
    return answer

dict(zip(name, yearning)) : name, yearning 배열을 하나의 dictionary로 만들기

zip : 두개의 리스트를 묶어줄 때 사용

해당 키 값이 딕셔너리에 있는지 확인 : if each_name in name_dict

'코딩테스트' 카테고리의 다른 글

[Python] 공원 산책  (0) 2024.03.12
[Python] 가장 가까운 같은 글자  (0) 2024.03.07
[Python] 문자열 겹쳐쓰기  (0) 2024.03.05
[Python] 달리기 경주  (0) 2024.03.05
[Python] 최댓값과 최솟값  (0) 2024.01.19