Algorithm / 대충 만든 자판

1) 문제

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

예를 들어, 1번 키에 "A", "B", "C" 순서대로 문자가 할당되어 있다면 1번 키를 한 번 누르면 "A", 두 번 누르면 "B", 세 번 누르면 "C"가 되는 식입니다.

같은 규칙을 적용해 아무렇게나 만든 휴대폰 자판이 있습니다. 이 휴대폰 자판은 키의 개수가 1개부터 최대 100개까지 있을 수 있으며, 특정 키를 눌렀을 때 입력되는 문자들도 무작위로 배열되어 있습니다. 또, 같은 문자가 자판 전체에 여러 번 할당된 경우도 있고, 키 하나에 같은 문자가 여러 번 할당된 경우도 있습니다. 심지어 아예 할당되지 않은 경우도 있습니다. 따라서 몇몇 문자열은 작성할 수 없을 수도 있습니다.

이 휴대폰 자판을 이용해 특정 문자열을 작성할 때, 키를 최소 몇 번 눌러야 그 문자열을 작성할 수 있는지 알아보고자 합니다.

1번 키부터 차례대로 할당된 문자들이 순서대로 담긴 문자열배열 keymap과 입력하려는 문자열들이 담긴 문자열 배열 targets가 주어질 때, 각 문자열을 작성하기 위해 키를 최소 몇 번씩 눌러야 하는지 순서대로 배열에 담아 return 하는 solution 함수를 완성해 주세요.

단, 목표 문자열을 작성할 수 없을 때는 -1을 저장합니다.

2) 문제 풀이

• Dictionary를 활용하여 각 알파벳에 대한 최소 입렷 횟수를 정의
• targets를 순회하며 최소 입력 횟수 저장

func solution(_ keymap:[String], _ targets:[String]) -> [Int] {
    var answer: [Int] = []
    var minPresses: [Character: Int] = [:]
    
    for key in keymap {
        for (i, char) in key.enumerated() {
            let pressCount = i + 1
            if let existring = minPresses[char] {
                minPresses[char] = min(existring, pressCount)
            } else {
                minPresses[char] = pressCount
            }
        }
    }
    
    for target in targets {
        var press: Int = 0
        for j in target {
            if let count = minPresses[j] {
                press += count
            } else {
                press = -1
                break
            }
        }
        answer.append(press)
    }
    
    return answer
}

결과

3) 코드 개선

• answer 배열을 미리 map을 사용해서 만들기

  • ➡️ 기존에는 for문을 사용하여 anwer.append(press)를 사용
  • ➡️ map을 사용하면 불필요한 변수 사용 없이 더 깔끔

• minPresses 딕셔너리 업데이트 과정 개선

  • ➡️ 기존에는 if let을 사용해 min(existring, pressCount)를 계산
  • ➡️ 더 간결하게 처리

• target을 순회할 때 reduce 활용

  • ➡️ for 루프 대신 reduce를 사용

func solution(_ keymap:[String], _ targets:[String]) -> [Int] {
    var minPresses: [Character: Int] = [:]

    // 🔹 keymap을 순회하며 최소 키 입력 횟수 저장
    for key in keymap {
        for (i, char) in key.enumerated() {
            minPresses[char] = min(minPresses[char] ?? Int.max, i + 1)
        }
    }

    // 🔹 targets 배열을 map을 이용해 변환 (더 깔끔한 방식)
    return targets.map { target in
        target.reduce(0) { (sum, char) in
            guard let pressCount = minPresses[char] else { return -1 }
            return sum == -1 ? -1 : sum + pressCount
        }
    }
}

결과