Lv.1 대충 만든 자판

문제 설명

휴대폰의 자판은 컴퓨터 키보드 자판과는 다르게 하나의 키에 여러 개의 문자가 할당될 수 있습니다. 키 하나에 여러 문자가 할당된 경우, 동일한 키를 연속해서 빠르게 누르면 할당된 순서대로 문자가 바뀝니다.

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

같은 규칙을 적용해 아무렇게나 만든 휴대폰 자판이 있습니다. 이 휴대폰 자판은 키의 개수가 1개부터 최대 100개까지 있을 수 있으며, 특정 키를 눌렀을 때 입력되는 문자들도 무작위로 배열되어 있습니다. 또, 같은 문자가 자판 전체에 여러 번 할당된 경우도 있고, 키 하나에 같은 문자가 여러 번 할당된 경우도 있습니다. 심지어 아예 할당되지 않은 경우도 있습니다. 따라서 몇몇 문자열은 작성할 수 없을 수도 있습니다.

이 휴대폰 자판을 이용해 특정 문자열을 작성할 때, 키를 최소 몇 번 눌러야 그 문자열을 작성할 수 있는지 알아보고자 합니다.

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

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

입출력

코드 풀이

def solution(keymaps,targets):
    data = dict()
    keylist = [] #이미 처리한 문자를 추적
    ans = []
    for key in keymaps:
        for k in key:
            if k not in keylist:
                data[k]=key.index(k)+1
                keylist.append(k)
            else:
                if key.index(k)+1<data[k]:
                    data[k]=key.index(k)+1
    for target in targets:
        tmp=0
        for t in target:
            if t not in keylist:
                tmp=-1
                break
            else:
                tmp+=data[t]
        ans.append(tmp)
    return ans

• 만약 k가 이미 keylist에 없다면, data 딕셔너리에 해당 문자(k)를 그 문자열에서의 위치와 함께 추가하고(인덱스 + 1), keylist에 해당 문자를 추가한다.
• 만약 k가 이미 keylist에 있다면, 현재 위치(인덱스 + 1)와 data 딕셔너리에 저장된 위치를 비교하여 더 작은 값을 data 딕셔너리에 업데이트한다.
• 만약 t가 keylist에 없다면, tmp 값을 -1로 설정하고 현재 문자열(target)에 대한 처리를 중단한다. 그렇지 않으면, data 딕셔너리에서 해당 문자(t)에 대응하는 값을 가져와서 tmp에 더한다.

출처: 프로그래머스 코딩테스트 연습