def solution(name, yearning, photo):
answer = []
# {'may': 5, 'kein': 10, 'kain': 1, 'radi': 3}
name_dict = {name:yearning[i] for i,name in enumerate(name)}
for i,row in enumerate(photo):
total = 0
for j,col in enumerate(row):
if name_dict.get(col) != None:
total += name_dict[col]
answer.append(total)
return answer
이번 문제는 사전 자료형을 사용하여 간단하게 풀 수 있던 문제였다. 여기서 기억해야 할 점은 사전 자료형에서 어떠한 값을 접근할 때 get() 메서드를 사용하지 않으면 에러가 발생하기 때문에 보다 안전하게 사전 자료형의 값에 접근하기 위해서는 get() 함수를 사용하여 해당 key를 가진 값이 존재하는지 확인하여야 한다.
'Algorithm 💡 > Implementation' 카테고리의 다른 글
[프로그래머스] 바탕화면 정리 (0) | 2023.06.23 |
---|---|
[프로그래머스] 공원 산책 (0) | 2023.06.23 |
[프로그래머스] 달리기 경주 (0) | 2023.06.23 |
[Softeer 기출문제] GBC (0) | 2023.05.30 |
[Softeer 기출문제] 전광판 (0) | 2023.05.30 |