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 |