자료구조&알고리즘/프로그래머스

[Python] 대충 만든 자판

두잇 두두 2024. 12. 22. 10:01
728x90

배경

 

 

코드

from collections import defaultdict


def solution(keymap, targets):
    answer = []
    MAX_INT = 101
    dic = defaultdict(lambda: MAX_INT)

    for kk in keymap:
        for i in range(len(kk)):
            dic[kk[i]] = min(dic[kk[i]], i + 1)

    for target in targets:
        cnt = 0
        for t in target:
            if t in dic:
                cnt += dic[t]
            else:
                cnt = - 1
                break
        answer.append(cnt)
    return answer

 

 

배운 점

defaultdict의 초기값을 lambda로 설정하는 법을 배움