문제
휴대폰의 자판은 컴퓨터 키보드 자판과는 다르게 하나의 키에 여러 개의 문자가 할당될 수 있습니다. 키 하나에 여러 문자가 할당된 경우, 동일한 키를 연속해서 빠르게 누르면 할당된 순서대로 문자가 바뀝니다.
예를 들어, 1번 키에 "A", "B", "C" 순서대로 문자가 할당되어 있다면 1번 키를 한 번 누르면 "A", 두 번 누르면 "B", 세 번 누르면 "C"가 되는 식입니다.
같은 규칙을 적용해 아무렇게나 만든 휴대폰 자판이 있습니다. 이 휴대폰 자판은 키의 개수가 1개부터 최대 100개까지 있을 수 있으며, 특정 키를 눌렀을 때 입력되는 문자들도 무작위로 배열되어 있습니다. 또, 같은 문자가 자판 전체에 여러 번 할당된 경우도 있고, 키 하나에 같은 문자가 여러 번 할당된 경우도 있습니다. 심지어 아예 할당되지 않은 경우도 있습니다. 따라서 몇몇 문자열은 작성할 수 없을 수도 있습니다.
이 휴대폰 자판을 이용해 특정 문자열을 작성할 때, 키를 최소 몇 번 눌러야 그 문자열을 작성할 수 있는지 알아보고자 합니다.
1번 키부터 차례대로 할당된 문자들이 순서대로 담긴 문자열배열 keymap과 입력하려는 문자열들이 담긴 문자열 배열 targets가 주어질 때, 각 문자열을 작성하기 위해 키를 최소 몇 번씩 눌러야 하는지 순서대로 배열에 담아 return 하는 solution 함수를 완성해 주세요.
단, 목표 문자열을 작성할 수 없을 때는 -1을 저장합니다
접근
해결 방식은 생각하기 어려운 편은 아니었다. 그러나 따져봐야 할 게 많아서 좀 복잡할까 싶어 차근차근 정리하면서 생각해봤다.
1. targets[0]
1-1. targets[0] 의 문자열을 하나씩 돌면서 keymap[0] 의 문자열에 몇번째에 있는지 판별
1-2. targets[0] 의 문자열을 하나씩 돌면서 keymap[1] 의 문자열에 몇번째에 있는지 판별
1-3. 을 keymap 의 길이만큼 반복하면서 -1이 아닐 경우에 리스트에 담는다.
1-4. 리스트의 최솟값을 반환해 answer 에 append 한다.
2. targets[1] 반복
3. 을 targets 의 길이만큼 반복
1차 시도
def solution(keymap, targets):
answer = []
for target in targets:
cnt_sum=0
for alphabet in target:
cnt_lst = []
for key in keymap:
cnt = key.find(alphabet)
if cnt >= 0:
cnt_lst.append(cnt+1)
if cnt_lst == []:
cnt_lst.append(-1)
cnt_sum += min(cnt_lst)
answer.append(cnt_sum)
return answer반복문 조건문이 마음에 안 들지만 우선 생각나는 대로 작성해보고 시도해보았다.
그러나
끼워맞추기를 시도한 기분으로 작성했더니 역시나 실패한 테스트케이스들이 있었다.
2차 시도
def solution(keymap, targets):
answer=[]
key_dict = {}
for i in range(len(keymap)):
for j in range(len(keymap[i])):
char = keymap[i][j]
if char not in key_dict:
key_dict[char] =(j + 1)
else:
key_dict[char] = min(key_dict[char],(j + 1))
for target in targets:
sum=0
for t in target:
if t in key_dict:
sum += key_dict[t]
else:
sum = -1
break
answer.append(sum)
return answer
GitHub 댓글