본문 바로가기

# 02/Swift - CTP

[Swift] 코딩테스트 연습! Lv1. 대충 만든 자판

반응형
import Foundation

/* 대충 만든 자판

 - 휴대폰의 자판은 컴퓨터 키보드 자판과는 다르게 하나의 키에 여러 개의 문자가 할당될 수 있습니다. 키 하나에 여러 문자가 할당된 경우, 동일한 키를 연속해서 빠르게 누르면 할당된 순서대로 문자가 바뀝니다.
 
 예를 들어, 1번 키에 "A", "B", "C" 순서대로 문자가 할당되어 있다면 1번 키를 한 번 누르면 "A", 두 번 누르면 "B", 세 번 누르면 "C"가 되는 식입니다.

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

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

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

 단, 목표 문자열을 작성할 수 없을 때는 -1을 저장합니다.
*/
func solution7(_ keymap:[String], _ targets:[String]) -> [Int] {
    var dic:[Character:Int] = [
        "A":-1,
        "B":-1,
        "C":-1,
        "D":-1,
        "E":-1,
        "F":-1,
        "G":-1,
        "H":-1,
        "I":-1,
        "J":-1,
        "K":-1,
        "L":-1,
        "M":-1,
        "N":-1,
        "O":-1,
        "P":-1,
        "Q":-1,
        "R":-1,
        "S":-1,
        "T":-1,
        "U":-1,
        "V":-1,
        "W":-1,
        "X":-1,
        "Y":-1,
        "Z":-1,
    ]
    for d in dic.keys {
        var i = 100
        for s in keymap {
            if let index = s.firstIndex(of: d) {
                let int_index = s.distance(from: s.startIndex, to: index)
                if i > int_index {
                    i = int_index
                }
            }
        }
        if i < 100 {
            dic[d] = i
        }
    }
    var result:[Int] = []
    for t in targets {
        var s = 0
        for c in Array(t) {
            if dic[c] == -1 {
                s = -1
                break
            }
            s += dic[c]!+1
        }
        result.append(s)
    }
    return result
}

solution7(["ABACD", "BCEFD"], ["ABCD","AABB"]) // [9, 4]
solution7(["AA"], ["B"]) // [-1]
반응형