[프로그래머스] LV.1 대충만든 자판

문제

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

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

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

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

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

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

제한사항

• 1 ≤ keymap의 길이 ≤ 100
• 1 ≤ keymap의 원소의 길이 ≤ 100
• keymap[i]는 i + 1번 키를 눌렀을 때 순서대로 바뀌는 문자를 의미합니다.
• 예를 들어 keymap[0] = "ABACD" 인 경우 1번 키를 한 번 누르면 A, 두 번 누르면 B, 세 번 누르면 A 가 됩니다.
• keymap의 원소의 길이는 서로 다를 수 있습니다.
• keymap의 원소는 알파벳 대문자로만 이루어져 있습니다.
• 1 ≤ targets의 길이 ≤ 100
• 1 ≤ targets의 원소의 길이 ≤ 100
• targets의 원소는 알파벳 대문자로만 이루어져 있습니다.

입출력 예시

keymap	             targets	       result
["ABACD", "BCEFD"]	["ABCD","AABB"]	   [9, 4]
["AA"]	            ["B"]	           [-1]
["AGZ", "BSSS"]	    ["ASA","BGZ"]	   [4, 6]

풀이

1. targets에 있는 문자열의 문자를 한 글짜씩 빼주어 keymap 리스트에 있는지 확인 해준 후
2. 없으면 -1을 리턴할 변수 answer에 저장해주고, 있다면 해당 키의 최소 클릭 수를 저장해준다.
3. 중복처리와 최소값 처리에 신경써야 하며 keymap에 문자가 없을경우 반복문에서 빠져나가기 위해 break를 꼭 해줘야한다.

def solution(keymap, targets):
    answer = []
    for index, target in enumerate(targets):
        cnt = 0
        answer.append(0)
        for t in target:
            key_temp = []
            if check_key(t, keymap) == True:
                # keymab 에 t가 잇음
                for key in keymap:
                    for i, k in enumerate(key):
                        if k == t:
                            key_temp.append(i+1)
                cnt+=min(key_temp)
                answer[index] = cnt    
            else:
                # keymab 에 t가 없음 -> answer배열에 append -1
                answer[index] = -1
                break # 붙여줘야함!
    return answer

def check_key(string, key_list):
    for key in key_list:
        if string in key:
            
            return True
    return False

• 먼저 targets 리스트를 enumerate()함수로 index와 함께 for 문으로 입력된 문자열들을 추출해준다.
• 매 요소마다 cnt를 0 으로 초기화해주는데 이 변수는 targets의 요소마다 클릭될 최소값을 의미하게 된다. 그리고 리턴해줄 변수 answer 리스트에도 0을 넣어주어 나중에 해당 요소의 값을 변경할 수 있게 해준다.
• 두 번째 for문은 targets에서 추출된 문자열을 한 문자씩 추출해주는데 key_temp 리스트를 정의하고 초기화하여 해당 문자가 keymap의 요소들에 있으면 모든 중복되는 문자들의 클릭수를 넣어주어 최소값을 측정할 수 있다. 또한 추출된 한 문자가 keymap변수에 속해 있는지 if문으로 확인을 먼저 해준다.
• 여기서는 새로운 함수 check_key(string, key_list)를 만들어 주어 사용하는데 이 함수는 매개변수로 문자와 문자열 리스트를 받는데 targets에서 추출된 문자열을 문자와 keymap을 매개변수로 넣어준다. 만약 해당 문자가 있다면 True를 아니면 False 를 반환한다.
• 만약 keymap에 해당 문자가 있다면 이중 for문을 통해 해당 문자가 몇번 째 위치에 있는지 확인을 해주는데 enumerate()함수를 다시 사용하여 클릭수를 측정할 수 있다. 해당 문자열과 keymap 문자열의 문자가 같으면 그 index값에 +1을 해주어 클릭수를 측정한 후 key_temp변수에 모든 클릭수를 넣어주도록 한다.
• 이제 다시 targets에서 한 문자를 뽑는 for문으로 돌아와서 초기화 되엇던 cnt변수에 key_temp에 있는 클릭 수 리스트에서 최소값을 더해준다. 그 후 answer변수 중 targets의 index에 해당하는 변수에 cnt값을 입력해준다. 이렇게 targets의 한 요소의 문자 개수만큼 for문이 진행되면 최소 클릭수를 모두 더한 값이 입력된다.
• 다음은 위 if문에서 해당 문자가 keymap에 없을 경우인데 이때는 해당 문자열을 keymap이 만들 수 없으므로 answer변수 중 targets의 index에 해당하는 변수에 -1 값을 넣준다. 그리고 break로 해당 반복문을 꼭 나와주어야 한다.

오답노트

• for 문이 여러개 사용되어 복잡하고 어디서 해당 값들을 입력해줄지 잘 고민해야한다.
• 그리고 keymap으로 targets의 요소를 만들지 못할 때(if 문의 else부분) break 를 꼭 해주어야한다. 이것때문에 시간을 좀 많이 소비했는데 원인은 간단하다. -1을 하고도 for문이 돌아갈 경우 해당 target이 만들어질 수 있는 것 처럼 될 수 있기 때문이다.