프로그래머스 '대충 만든 자판' 파이썬 풀이

프로그래머스

"""
휴대폰의 자판은 컴퓨터 키보드 자판과는 다르게
하나의 키에 여러 개의 문자가 할당될 수 있습니다.

키 하나에 여러 문자가 할당된 경우, 동일한 키를 연속해서 빠르게 누르면
할당된 순서대로 문자가 바뀝니다.

예를 들어, 1번 키에 "A", "B", "C" 순서대로 문자가 할당되어 있다면
1번 키를 한 번 누르면 "A", 두 번 누르면 "B", 세 번 누르면 "C"가 되는 식입니다.
(중략)
"""

def solution(keymap, targets):
    answer = []
    kv = {}
    for keys in keymap:
        for index, key in enumerate(keys):
            if key in kv.keys():
                kv[key] = min(index+1, kv[key])
            else:
                kv[key] = index+1
                
    for target in targets:
        try:
            answer.append(sum([kv[key] for key in target]))
        except:
            answer.append(-1)
            
    return answer
    
"""
- 15 line
각 알파벳을 최소로 누른 횟수를 저장하기 위해 
kv (key, value)라고 dict를 만들었다. 

- 16 line
dict에 넣기 위해 keymap에서 문자열을 하나씩 가지고 온다.

- 17 line
문자열에서 하나씩 문자를 가지고 오는데, 
index+1번만큼 눌러야 하기때문에 enumerate로 문자위치도 체크한다.

- 18~21 line
만약 kv에 문자(key)의 값(index+1)이 들어가 있다면
기존의 들어있는 값과 지금 값을 비교하여 더 작은값을 다시 문자의 값으로 지정한다.
들어가있지 않다면 현재 값을 입력한다.

- 23~ line
이제 targets의 문자열을 가지고 오면서 answer에 값을 추가시켜준다.
먼저 target문자열에서 문자 하나씩을 읽어와서 kv에서 문자에 해당하는 값을 추가시킨다.

예를 들어서 kv = {'A': 1, 'B': 2} 이고 target문자열이 'AB'이라면
[kv[key] for key in target]은 [1, 2]가 된다. 이값을 sum하면 AB의 입력값이 된다.

근데 만약에 kv에 값이 입력되어있지 않는 문자라면 try, except문에서 걸러지게 된다.
그럼 -1을 입력한다.

여담으로 제한사항이 너무 이상한 것 같다.
그래서 질문으로 남겨두었는데 답변이 왔으면 좋겠다.
https://school.programmers.co.kr/questions/44822
"""