문제
문제 : https://school.programmers.co.kr/learn/courses/30/lessons/160586
컴퓨터 키보드 자판과 다르게 휴대폰의 자판은 하나의 키에 여러 개의 문자가 할당될 수 있다. 키 하나에 여러 문자가 할당된 경우, 동일한 키를 연속해서 빠르게 누르면 할당된 순서대로 문자가 바뀐다.
예를 들어, 1번 키에 "A", "B", "C" 순서대로 문자가 할당되어 있다면 1번 키를 한 번 누르면 "A", 두 번 누르면 "B", 세 번 누르면 "C"가 된다.
동일한 규칙으로 만든 휴대폰 자판이 있다고 하자. 이 휴대폰 자판은 키가 1개부터 최대 100개까지 있을 수 있으며, 특정 키를 눌렀을 때 입력되는 문자들도 무작위로 배열되어 있다. 같은 문자가 자판 전체에 여러 번 할당된 경우도 있고, 키 하나에 같은 문자가 여러 번 할당된 경우도 있다. 심지어 아예 할당되지 않은 경우도 있다.
이 휴대폰 자판을 이용해 특정 문자열을 작성할 때, 문자열을 작성할 수 있는 최소 키 입력 횟수를 알아보고자 한다.
1번 키부터 차례대로 할당된 문자들이 순서대로 담긴 문자열배열 keymap과 입력하려는 문자열들이 담긴 문자열 배열 targets가 주어질 때, 각 문자열을 작성하기 위한 최소 키 입력 수를 배열에 담아 반환한다.
단, 목표 문자열을 작성할 수 없을 때는 -1을 저장합니다.
풀이
키에 할당되어 있는 문자의 최소 입력 수를 저장하고, 문자열 작성에 필요한 입력 수의 합을 구하면 된다.
1. 키에 할당되어 있는 문자의 최소 입력 수 저장
각 키의 입력 수를 저장해야 하므로 Dictionary를 사용하면 될 것 같다. key_cnt라는 변수에 저장한다. 방법은 다음과 같다.
- 각 키의 할당된 문자열을 읽는다.
- 문자가 key_cnt에 저장되어 있지 않으면 필요한 입력 수를 저장한다.
- 문자가 이미 key_cnt에 저장되어 있으면 기존 입력 수와 현재 입력 수 중 작은 값으로 갱신한다.
- 각 키에 할당된 문자열만큼, 키의 개수만큼 반복한다.
key_cnt = dict()
for key in keymap:
for i in range(len(key)):
key_cnt[key[i]] = min(key_cnt[key[i]], i) if key[i] in key_cnt.keys() else i
2. 문자열 작성에 필요한 입력 수의 합 계산
- 문자열을 구성하는 각 문자를 key_cnt에서 찾아 그 입력 수의 합을 구한다.
- key_cnt에 없는 값이 있으면 -1로 저장한다.
- 입력할 문자열의 수만큼 반복한다.
answer = []
for target in targets:
try:
answer.append(sum([key_cnt[t] + 1 for t in target]))
except KeyError:
answer.append(-1)
return answer
전체 코드는 접은글로 작성해 둔다.
def solution(keymap, targets):
key_cnt = dict()
for key in keymap:
for i in range(len(key)):
key_cnt[key[i]] = min(key_cnt[key[i]], i) if key[i] in key_cnt.keys() else i
answer = []
for target in targets:
try:
answer.append(sum([key_cnt[t] + 1 for t in target]))
except KeyError:
answer.append(-1)
return answer
참고 문서
https://school.programmers.co.kr/learn/courses/30/lessons/160586