문제 설명

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

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

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

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

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

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

문제 해결

def solution(keymap, targets):
    answer = []
    for i in targets: #target 각 원소 접근
        Sum = 0
        for j in i: #target 각 원소의 문자 하나 접근
            arr = []
            for k in keymap: #k번째 keymap에 접근
                if j in k:
                    arr.append(k.index(j) + 1) #해당하는 키의 인덱스 + 1
            if not arr: #해당 단어를 찾지 못했을때
                Sum = 0
                break
            else: #해당 단어를 찾았을 때 각 keymap의 원소 중 최소 인덱스 값을 합해줌
                Sum += min(arr)
        if Sum == 0: #만약 총 합이 0이면 찾지 못했으므로 -1을 넣어줌
            answer.append(-1)
        else:
            answer.append(Sum)
                
    return answer

문제 풀이 설명

1. 첫번째 풀이

def solution(keymap, targets):
    answer = []
    for i in targets: #target 각 원소 접근
        total = 0
        for j in i: #target 각 원소의 문자 하나 접근
            for k in keymap: #k번째 keymap에 접근
                if j in k:
                    total += (k.index(j) + 1) #해당하는 키의 인덱스 + 1
                    break #k문 탈출
        if total == 0:
            answer.append(-1)
        else:
            answer.append(total)
    return answer

keymap을 순서대로 접근해서 테스트케이스 오류가 발생했다.

=> 해결방법으로 각 글자를 찾은 인덱스를 stack처럼 쌓은 뒤 최솟값을 합하자는 생각!

2. 두번째 풀이

def solution(keymap, targets):
    answer = []
    for i in targets: #target 각 원소 접근
        Sum = 0
        for j in i: #target 각 원소의 문자 하나 접근
            arr = []
            for k in keymap: #k번째 keymap에 접근
                if j in k:
                    arr.append(k.index(j) + 1) #해당하는 키의 인덱스 + 1
            if not arr: #해당 단어를 찾지 못했을때
            	continue
            else: #해당 단어를 찾았을 때 각 keymap의 원소 중 최소 인덱스 값을 합해줌
                Sum += min(arr)
        if Sum == 0:
            answer.append(-1)
        else:
            answer.append(Sum)
                
    return answer

반례 ["ABACD", "BCEFD"], ["XABCD", "AABB"], 리턴값 [-1, 4] 에 오류 발생

=> 해당 단어를 찾지 못했을 때 continue로 넘기지 말고 break로 변경, -1이 들어갈 수 있도록 수정함

하지만, 그래도 테케 14~23까지 오류 발생..

반례 ["AGB", "BSSS"], ["AGZ", "BSSS"], 리턴값[-1, 7] 에 오류 발생인 걸 발견

=> 처음 단어가 아닌 이후 없는 단어를 발견 시 -1이 들어가지 않는 것을 알고 단순히 break가 아닌 Sum 변수를 0으로 초기화 해줌으로써 단어를 발견하지 못한 걸 인지시켰더니 해결됐다!