파이썬 알고리즘 : 대충 만든 자판

Dictionary

2024년 3월 25일 알고리즘 문제풀이

문제

난이도

Lv.1

코드

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
def solution(keymap, targets):
    answer = []
    arr = dict()
    idx = 0
    
    # 각 글자를 입력하기 위한 버튼 누르는 횟수의 최솟값을 저장하는 배열 arr을 만드는 과정
    # 각 버튼의 0번 index 부터 하나씩 참조
    while True:
        # 모든 키의 입력 범위를 넘어갔는지 확인하기 위한 플래그
        flag = False
        # 모든 버튼 순회
        for button in keymap:
            # 예를 들어, 버튼을 5번 누를 수 있는데 idx가 6이라면 그 버튼은 더이상 조사할 필요가 없으므로 넘어감
            if idx >= len(button):
                continue
            # 어떤 버튼을 참조했음을 플래그에 표시. 이 과정이 없으면 keymap중 가장 긴 버튼의 길이를 구하느 과정이 따로 필요
            flag = True
            # idx번 눌렀을 때 해당 버튼에서 입력되는 글자
            letter = button[idx]
            # 지금 입력되는 글자가, 아직 입력된적 없는 글자라면 배열에 추가
            if letter not in arr:
                arr[letter] = idx+1
            # 이미 입력된적 있다면, idx를 작은 값으로 바꿔줌. 각 글자를 누르는 최소 방법만 알면 되기 때문
            # 0번 인덱스는 1번 눌러야 하므로 idx+1을 저장
            else:
                arr[letter] = min(arr[letter],idx+1)
        # 아무 버튼도 탐색하지 않았다면 무한루프 종료
        if not flag:
            break
        idx += 1
        
    for target in targets:
        cnt = 0
        # 타겟을 하나씩 탐색하며 글자를 확인
        for i in range(len(target)):
            # 만약 글자를 입력할 방법이 있다면 해당 키의 값을 추가
            if target[i] in arr:
                cnt += arr[target[i]]
            # 글자를 입력할 방법이 없다면 -1로 지정하고 종료
            else:
                cnt = -1
                break
        answer.append(cnt)
    return answer