[ 프로그래머스 ] 대충 만든 자판

 

 

🍄 코딩테스트

코딩테스트 문제 풀이

✍🏻 Github

문제 풀이 github 링크

 

 

문제 설명

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

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

 

입출력 예

keymap	targets	result
["ABACD", "BCEFD"]	["ABCD","AABB"]	[9, 4]
["AA"]	["B"]	[-1]
["AGZ", "BSSS"]	["ASA","BGZ"]	[4, 6]

 

입출력 예 설명

입출력 예 #1

• "ABCD"의 경우,
• 1번 키 한 번 → A
• 2번 키 한 번 → B
• 2번 키 두 번 → C
• 1번 키 다섯 번 → D
• 따라서 총합인 9를 첫 번째 인덱스에 저장합니다.
• "AABB"의 경우,
• 1번 키 한 번 → A
• 1번 키 한 번 → A
• 2번 키 한 번 → B
• 2번 키 한 번 → B
• 따라서 총합인 4를 두 번째 인덱스에 저장합니다.
• 결과적으로 [9,4]를 return 합니다.

입출력 예 #2

• "B"의 경우, 'B'가 어디에도 존재하지 않기 때문에 -1을 첫 번째 인덱스에 저장합니다.
• 결과적으로 [-1]을 return 합니다.

입출력 예 #3

• "ASA"의 경우,
• 1번 키 한 번 → A
• 2번 키 두 번 → S
• 1번 키 한 번 → A
• 따라서 총합인 4를 첫 번째 인덱스에 저장합니다.
• "BGZ"의 경우,
• 2번 키 한 번 → B
• 1번 키 두 번 → G
• 1번 키 세 번 → Z
• 따라서 총합인 6을 두 번째 인덱스에 저장합니다.
• 결과적으로 [4, 6]을 return 합니다.

 

 

문제 풀이

나의 풀이

import Foundation

func solution(_ keymap:[String], _ targets:[String]) -> [Int] {
    // 문자의 최소 클릭 수
    var clickOfChar = Dictionary<Character, Int>()
    var count: Int = 0
    var answer = [Int]()
    // 문자의 최소 클릭수 찾기
    func searchMinClick(_ alphabet: Character) -> Int {
        for key in keymap {
            if key.contains(alphabet) {
                let value = Array(key).firstIndex(of: alphabet)! + 1
                if value < clickOfChar[alphabet] ?? -1 || clickOfChar[alphabet] == nil {
                     clickOfChar[alphabet] = value
                }
            } 
        }
        return clickOfChar[alphabet] ?? -1
    }
    // 문자열의 최소 클릭수 구하기
    for str in targets {
        for char in str {
            if let c = clickOfChar[char] {
                count += c
            }else{
                let s = searchMinClick(char)
                if s == -1 {
                    count = -1
                    break
                }else {
                    count += s
                }
            }
        }
        answer.append(count)
        count = 0
    }
    return answer
}

 

우선 무조건 keymap과 targets을 탐색해야한다는 생각에

시간복잡도를 어떻게 하면 줄일 수 있을까를 생각했던 것 같다.

 

나는 딕셔너리에 문자와 그 문자의 최소 클릭수를 넣기로 했다.

그럼 targets를 탐색하여 문제를 풀 때, 이미 추론했던 문자라면 또 다시 반복문을 돌지 않아도 되니깐

만약 “ABC”문자열의 최소 클릭수를 찾고자 할 때,

“A”라는 문자를 탐색해야하는데 이미 앞에서 “ERTA”문자열을 탐색하여

딕셔너리에 “A”의 최소 클릭 수가 1이라고 추론을 했다면

다시 반복문을 돌면 최소 클릭 수를 찾지 않아도 된다.

 

이렇게 풀기 위해서는 해당 문자에 대한 최소 클릭수를 구하는 부분이 따로 구현이 되어야 했다.

그래서 함수안에 함수로 구현을 하였다.

 

딕셔너리에 최소 문자 클릭수를 대입하기 위해서는

우선 해당 문자열에 문자가 들어있는지를 판단을 하고

문자열이 들어있다면, 몇 번째 문자열인지를 배열의 인덱스를 활용해 구했다.

구한 인덱스가 이미 딕셔너리에 있는지 없는지를 판단하고

만약 있다면 값비교를 통해 더 작은 클릭수를 업데이트 하도록 구현을 하였다.

 

 

자판.. 진짜 대충 만들었네..