본문 바로가기
문제풀이/프로그래머스

[Python]대충 만든 자판

by PhoB 2023. 5. 9.

[문제]

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

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

[내 코드]

import math

def solution(keymap, targets):
    answer = []
    dic = {}
    
    for word in keymap:
        for i,letter in enumerate(word):
            if letter not in dic:
                dic[letter] = i+1
            else:
                dic[letter] = min(dic[letter],i+1)
    
    for target in targets:
        cnt = 0
        for key in target:
            if key not in dic:
                cnt = -1
                break
            cnt += dic[key]
        answer.append(cnt)
        
    return answer

[설명]

딕셔너리를 활용한 풀이입니다. 이 문제는 target단어 완성까지 최소한으로 버튼을 누르는 경우의 수를 구하는 문제입니다. 즉 원하는 알파벳을 가장 적게 눌러서 출력할 수 있는 keymap을 찾아야합니다. 하지만 이렇게하면 타겟 단어의 글자 하나하나마다 keymap의 갯수만큼의 반복문과 또 각각의 keymap안을 또 반복하며 심지어 최소값 검사도 해줘야해서 때문에 굉장히 비효율적입니다.

이해를 돕기위한 예시...입니다..ㅎㅎ

따라서 딕셔너리를 활용하여 그 과정을 최소화 하는것 입니다. 또한 enumerate를 사용하여 for문 하나로 [key,value]튜플 하나를 반환하도록 하였습니다. 딕셔너리에 없는 알파벳이면 새로 하나 만들어주고, 존재하는 값이면 기존값과 비교하여 더 적은 값을 넣도록 하였습니다

알파벳[key] 최소 횟수[value]
A 1
B 1
C 2

이후는 딕셔너리에 기록된 데이터들을 토대로 key에 상응하는 value를 가져와 마무리 해주시면 됩니다!

 

'문제풀이 > 프로그래머스' 카테고리의 다른 글

[Python]신규 아이디 추천  (1) 2023.05.13
[Python]키패드 누르기  (0) 2023.05.11
[Python] 크레인 인형뽑기 게임  (0) 2023.05.09
[Python] 완주하지 못한 선수  (0) 2023.05.02
[Python] 체육  (0) 2023.05.02