Algorithm / 둘만의 암호

1) 문제

두 문자열 s와 skip, 그리고 자연수 index가 주어질 때, 다음 규칙에 따라 문자열을 만들려 합니다. 암호의 규칙은 다음과 같습니다.

문자열 s의 각 알파벳을 index만큼 뒤의 알파벳으로 바꿔줍니다.
index만큼의 뒤의 알파벳이 z를 넘어갈 경우 다시 a로 돌아갑니다.
skip에 있는 알파벳은 제외하고 건너뜁니다.
예를 들어 s = "aukks", skip = "wbqd", index = 5일 때, a에서 5만큼 뒤에 있는 알파벳은 f지만 [b, c, d, e, f]에서 'b'와 'd'는 skip에 포함되므로 세지 않습니다. 따라서 'b', 'd'를 제외하고 'a'에서 5만큼 뒤에 있는 알파벳은 [c, e, f, g, h] 순서에 의해 'h'가 됩니다. 나머지 "ukks" 또한 위 규칙대로 바꾸면 "appy"가 되며 결과는 "happy"가 됩니다.

두 문자열 s와 skip, 그리고 자연수 index가 매개변수로 주어질 때 위 규칙대로 s를 변환한 결과를 return하도록 solution 함수를 완성해주세요.

2) 문제 풀이

func solution(_ s:String, _ skip:String, _ index:Int) -> String {
    var alphabets: [String] = [
        "", "a", "b" , "c", "d", "e", "f", "g", "h", "i", "j", "k", "l", "m", "n", "o", "p", "q", "r", "s", "t", "u", "v", "w", "x", "y", "z"
    ]
    var answer: [String] = []
    
    for alphabet in skip where alphabets.contains(String(alphabet)) {
        let skipWord = alphabets.firstIndex(of: String(alphabet))!
        alphabets.remove(at: skipWord)
    }
    
    for i in s where alphabets.contains(String(i)) {
        let wordIndex = alphabets.firstIndex(of: String(i))!
        let result = wordIndex + index >= alphabets.count ? (wordIndex + index) - (alphabets.count - 1) : wordIndex + index
        
        answer.append(alphabets[result])
    }
    
    return answer.joined()
}

결과

3) 코드 개선

• alphabets 배열 수정
• ! 를 사용한 옵셔널 언래핑 삭제
• 순환 로직 수정 -> (wordIndex + index) % alphabets.count
• 문자열 변환 과정 줄이기

func solution(_ s: String, _ skip: String, _ index: Int) -> String {
    // 1. 알파벳 중 skip 문자 제거
    let fullAlphabets = Array("abcdefghijklmnopqrstuvwxyz")
    let skipSet = Set(skip)
    let availableAlphabets = fullAlphabets.filter { !skipSet.contains($0) }

    // 2. 알파벳을 [Character: Int] 딕셔너리로 맵핑하면 더 빠르게 접근 가능
    let count = availableAlphabets.count
    var result = ""

    for char in s {
        if let currentIndex = availableAlphabets.firstIndex(of: char) {
            let newIndex = (currentIndex + index) % count
            result.append(availableAlphabets[newIndex])
        }
    }

    return result
}

✅ 개선된 점 요약

항목	개선 전	개선 후
알파벳 배열	수동 + 인덱스 1 보정 필요	"a"..."z" 배열 자동 생성
문자열 처리	String(i) 변환 반복	Character 그대로 처리
순환 처리	(i + index) - (count - 1) ← 불완전	(i + index) % count ← 안정적
안전성	! 강제 언래핑	옵셔널 안전 처리 (if let)
성능	firstIndex(of:) 반복 호출	개선 가능 (추가적으로 딕셔너리화 고려 가능)

결과