대충 만든 자판

from collections import defaultdict

def solution(keymap, targets):
    d = defaultdict(list)​

프로그래머스

 

 

문제

---

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

예를 들어, 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의 원소는 알파벳 대문자로만 이루어져 있습니다.

 

 

 

 

코드

---

from collections import defaultdict

def solution(keymap, targets):
    d = defaultdict(list)
    for key in keymap:
        for idx, char in enumerate(key):
            d[char].append(idx)
    
    min_keymap = {key: min(value)+1 for key, value in d.items()}
    
    answer = []
    for target in targets:
        cnt = 0
        for x in target:
            if x not in min_keymap.keys():
                cnt = -1
                break
            else:
                cnt += min_keymap.get(x)
        answer.append(cnt)
    
    return answer

 

 

 

 

노트

---

1. 알파벳 입력에 필요한 최소 횟수를 딕셔너리로 정리

from collections import defaultdict

def solution(keymap, targets):
    d = defaultdict(list)
    for key in keymap:
        for idx, char in enumerate(key):
            d[char].append(idx)
    
    min_keymap = {key: min(value)+1 for key, value in d.items()}
    
  	print(min_keymap)
    
'''
input
["ABACD", "BCEFD"], ["ABCD", "AABB"]

output
{'A': 1, 'B': 1, 'C': 2, 'D': 5, 'E': 3, 'F': 4}
'''

• defaultdict: 딕셔너리의 value에 들어갈 요소를 자동으로 설정해주는 딕셔너리
• d[key].append(value) 를 사용할 수 있음
• 알파벳과 알파벳을 입력하기 위해 눌러야하는 최소횟수를 min_keymap 딕셔너리로

 

2. 반복문으로 횟수 카운트 

    answer = []
    for target in targets:  # (1)번 반복문
        cnt = 0
        for x in target:
            if x not in min_keymap.keys():  # (2)번 반복문
                cnt = -1
                break
            else:
                cnt += min_keymap.get(x)
        answer.append(cnt)
    
    return answer

• targets의 단어들을 하나씩 받아와서 구성 문자열을 입력하기 위한 횟수를 min_keymap 딕셔너리에서 찾아 세어 줌
• 만약 입력 불가한 단어가 나온다면 cnt변수에 -1을 할당하고 (2)번 반복문을 종료해 answer에 값 추가 

 

    hs = {}
    for k in keymap:
        for i, ch in enumerate(k):
            hs[ch] = min(i + 1, hs[ch]) if ch in hs else i + 1

• min_keymap을 만드는 천재적인 코드....
• 문자가 딕셔너리의 key(알파벳)에 없으면 키와 값(입력에 필요한 횟수)을 추가 
• key에 있으면 기존 값과 새로운 값 중 작은값을 대체 

 

 

참조

---

[파이썬 기초] 유사 딕셔너리 defaultdict() 활용법

프로그래머스