코딩테스트
[Python] 추억 점수
snoony
2024. 3. 5. 12:16
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