문제 설명
휴대폰의 자판은 컴퓨터 키보드 자판과는 다르게 하나의 키에 여러 개의 문자가 할당될 수 있습니다. 키 하나에 여러 문자가 할당된 경우, 동일한 키를 연속해서 빠르게 누르면 할당된 순서대로 문자가 바뀝니다.
예를 들어, 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]풀이
def solution(keymap, targets):
result = []
for target in targets :
count = 0
for char in target :
diff = 100
for key in keymap :
if char in key :
diff = min(key.index(char) + 1, diff)
if diff == 101:
count = -1
break
else :
count += diff
result.appen(count)
return result
키맵에서 저장하면서 만들어지는 코드가 궁금하여 생각해 보았다.
def solution_with_state(keymap, targets):
result = [] # 각 타겟 문자열에 대한 결과를 저장할 리스트
for target in targets: # 타겟 문자열 하나씩 처리
total_presses = 0 # 해당 문자열을 만들기 위한 총 키 입력 횟수
current_positions = [0] * len(keymap) # 각 키의 현재 탐색 위치를 저장
for char in target: # 타겟 문자열의 각 문자를 처리
found = False # 문자를 찾았는지 여부
min_presses = float('inf') # 현재 문자를 만들기 위한 최소 입력 횟수
for key_index, keys in enumerate(keymap): # 각 키맵을 탐색
if char in keys: # 현재 키에서 문자를 찾을 수 있는 경우
start_position = current_positions[key_index] # 현재 키의 상태
# 원형 탐색: 현재 위치 이후로부터 문자를 찾음
circular_keys = keys[start_position:] + keys[:start_position]
press_count = circular_keys.find(char) + 1
if press_count > 0: # 문자를 찾았다면
found = True # 찾음
if press_count < min_presses: # 최소 키 입력 횟수 갱신
min_presses = press_count
# 현재 키 상태를 갱신 (순환 형태)
current_positions[key_index] = (start_position + press_count) % len(keys)
if not found: # 문자를 어느 키에서도 찾을 수 없으면
total_presses = -1
break # 더 이상 처리할 필요 없음
total_presses += min_presses # 최소 키 입력 횟수를 누적
result.append(total_presses) # 해당 타겟 문자열의 결과를 저장
return result