[잘못누른자판]
휴대폰의 자판은 컴퓨터 키보드 자판과는 다르게 하나의 키에 여러 개의 문자가 할당될 수 있습니다. 키 하나에 여러 문자가 할당된 경우, 동일한 키를 연속해서 빠르게 누르면 할당된 순서대로 문자가 바뀝니다.
예를 들어, 1번 키에 "A", "B", "C" 순서대로 문자가 할당되어 있다면 1번 키를 한 번 누르면 "A", 두 번 누르면 "B", 세 번 누르면 "C"가 되는 식입니다.
같은 규칙을 적용해 아무렇게나 만든 휴대폰 자판이 있습니다. 이 휴대폰 자판은 키의 개수가 1개부터 최대 100개까지 있을 수 있으며, 특정 키를 눌렀을 때 입력되는 문자들도 무작위로 배열되어 있습니다. 또, 같은 문자가 자판 전체에 여러 번 할당된 경우도 있고, 키 하나에 같은 문자가 여러 번 할당된 경우도 있습니다. 심지어 아예 할당되지 않은 경우도 있습니다. 따라서 몇몇 문자열은 작성할 수 없을 수도 있습니다.
이 휴대폰 자판을 이용해 특정 문자열을 작성할 때, 키를 최소 몇 번 눌러야 그 문자열을 작성할 수 있는지 알아보고자 합니다.
1번 키부터 차례대로 할당된 문자들이 순서대로 담긴 문자열배열 keymap과 입력하려는 문자열들이 담긴 문자열 배열 targets가 주어질 때, 각 문자열을 작성하기 위해 키를 최소 몇 번씩 눌러야 하는지 순서대로 배열에 담아 return 하는 solution 함수를 완성해 주세요.
단, 목표 문자열을 작성할 수 없을 때는 -1을 저장합니다.
keymap | targets | result |
---|---|---|
["ABACD", "BCEFD"] | ["ABCD","AABB"] | [9, 4] |
["AA"] | ["B"] | [-1] |
["AGZ", "BSSS"] | ["ASA","BGZ"] | [4, 6] |
def solution(keymap, targets):
answer = []
dict = {}
for target in targets: # for each word target
res = 0
for char in target: # for each char in target
if char in dict.keys():
res += dict[char]
else: # if char not in dict.keys() = not been detected yet
for key in keymap: # each key, such as "ABACD"
for idx, char in enumerate(key): # (0,A), (1,B)
if char not in dict.keys():
dict[char] = idx + 1
res += dict[char]
print(char, res)
answer.append(res)
return answer
처음에 뭔가 잘못했던 코드
왜 잘못됐을까요..
그것은 바로 key의 각 char 중에 만약에 dictionary에 assign되지 않은 키에 대한 케이스를 다루지 않았기 때문입니다.... 바보
물론 문제는 그것 뿐만이 아니라 char 가 이미 dictionary에 들어가있을때 새로 들어오는 키와 값을 비교하지 않았다는 것도..! ㅋㅋㅋㅋ.ㅋ..
for key in keymap: # each key, such as "ABACD"
for idx, char in enumerate(key): # (0,A), (1,B)
if char not in dict.keys():
dict[char] = idx + 1
res += dict[char]
이렇게 했었는데 바로 res 에 더할게 아니라 체크를 먼저 한 다음에 진행했었어야했어요..멍청이
근데 이걸 막상 문제 풀때는 해결을 못해서 뭐지? 코드가 잘못됐나? 하고 한참을 생각하다가 코드를 새로 짜고서야 드디어 찾아냈습니다
def solution(keymap, targets):
answer = []
dict = {}
for target in targets:
res = 0
# update dictionary
for key in keymap:
for idx, char in enumerate(key):
if char not in dict.keys():
dict[char] = idx+1
else:
# if char already in dict, compare the value from the other key
# assign the minimum value
dict[char] = min(dict[char], idx+1)
# find the value from the dictionary
for char in target:
if char in dict.keys():
res += dict[char]
else:
res = -1
break
answer.append(res)
return answer
원래 생각했던 방법은 dictionary를 업데이트 해가면서 바로 값을 더하는식으로 (dp?라고 내 나름대로 생각..) 하는거였는데 예상치 못하게 오류가 생겨서 아예 dictionary를 완성한 후에 그걸 바탕으로 값을 더해가는 식으로 바꿨다
시간복잡도는 비슷한 것 같아요