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

프로그래머스

"""
휴대폰의 자판은 컴퓨터 키보드 자판과는 다르게 하나의 키에 여러 개의 문자가 할당될 수 있습니다. 
키 하나에 여러 문자가 할당된 경우, 동일한 키를 연속해서 
빠르게 누르면 할당된 순서대로 문자가 바뀝니다.
예를 들어, 1번 키에 "A", "B", "C" 순서대로 문자가 할당되어 있다면 1번 키를 
한 번 누르면 "A", 두 번 누르면 "B", 세 번 누르면 "C"가 되는 식입니다.

1번 키부터 차례대로 할당된 문자들이 순서대로 담긴 문자열배열 keymap과 
입력하려는 문자열들이 담긴 문자열 배열 targets가 주어질 때, 
각 문자열을 작성하기 위해 키를 최소 몇 번씩 눌러야 하는지 
순서대로 배열에 담아 return 하는 solution 함수를 완성해 주세요.

단, 목표 문자열을 작성할 수 없을 때는 -1을 저장합니다.
"""

def solution(keymap, targets):
    answer = []
    for target in targets:
        count, isTrue = 0, True
        for key in target:
            min_index = min([101] + [k.find(key)+1 for k in keymap if k.find(key)+1 != 0])
            if min_index == 101:
                isTrue = False
                break
            count += min_index
        answer.append(count if isTrue else -1)
        
    return answer

"""
keymap에 해당 알파벳이 있으면 리스트에 넣고 없으면 넣지 않아야겠다고 생각했다.
그리고 최소값이 기본값인 101이라면 keymap에 없다는 뜻이므로 -1을 넣어주었다.

먼저 targets리스트에서 target을 하나씩 가지고 오고, 
target의 키를 누를 횟수를 저장할 count, 
keymap에 알파벳이 있는지 확인하는 isTrue변수를 선언했다.

그리고 target에서 알파벳 하나씩 keymap에 있는지 판단하기 위해 for문을 돌렸다.
알파벳이 있는 가장 작은 인덱스를 찾아야하기 때문에 min함수를 사용했고
리스트 내포구문을 사용하여 keymap에서 인덱스를 찾아주었다.

우선 k.find(key)+1은 find함수를 사용하였고 find는 index함수와 비슷하다.
find함수는 문자열에서 key값을 찾아 인덱스값을 리턴하는 함수이고, 
만약 key가 없다면 -1을 리턴한다.

내포구문을 해석하면 keymap에서 하나씩 값을 가지고오는데, k.find(key)+1이 0이 아니라면
다시말해 keymap에서 내가 찾을 key가 키맵에 있다면 리스트에 추가한다는 뜻이다.
만약 keymap의 문자열 전체에 하나도 없다면 min값은 101이 된다. (하나도 추가하지 않았으니까)

그러면 keymap에서 인덱스를 찾을수 없으므로 isTrue를 false로 바꿔준다.
그래서 정답리스트에 인덱스를 다 찾을수있었으면 count값을 넣고,
아니라면 -1을 넣어서 찾을수 없다는 표시를 해준다.
"""