[프로그래머스] 대충 만든 자판 - Python

해당 문제는 프로그래머스 기준 Level 1에 해당하는 문제다. 

 

풀이는 2중 for문을 통해서 먼저 딕셔너리를 생성한다. 이때 자판 패드가 가장 가까운 번호를 기준으로 key와 value를 지정하면 된다. 

from collections import defaultdict
def solution(keymap, targets):
    answer = []
    inform = defaultdict(list)
    for key in keymap:
        for i, k in enumerate(key):
            if inform[k] == []:
                inform[k] = i+1
            
            elif inform[k] != [] and (i+1) < inform[k]:
                inform[k] = i+1
    

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

그런 다음, target을 기준으로 자판을 누르는 횟수가 가장 낮은 값으로 딕셔너리를 구성한 값을 바탕으로 합계를 더해주면 된다. if 문을 설정해두어서 키패드가 없는 경우에는 -1을 반환하여 속도를 조금 향상시켰다.