[프로그래머스] Swift - 둘만의 암호 (LV.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 함수를 완성해주세요.

 

제한사항

• 5 ≤ s의 길이 ≤ 50
• 1 ≤ skip의 길이 ≤ 10
• s와 skip은 알파벳 소문자로만 이루어져 있습니다.
  • skip에 포함되는 알파벳은 s에 포함되지 않습니다.
• 1 ≤ index ≤ 20

 

입출력 예제

s	skip	index	result
"aukks"	"wbqd"	5	"happy"

 

풀이 [ 메모리: 16.5mb, 최대시간:  1.66ms ] 

func solution(_ s:String, _ skip:String, _ index:Int) -> String {
    let alphabet = "abcdefghijklmnopqrstuvwxyz".map { String($0) }
    var s = s.map { String($0) }
    
    for i in 0..<s.count {
        if var c = alphabet.firstIndex(of: s[i]) {
            var index = index
            
            while index != 0 {      
                c += 1
                c %= 26
                
                if skip.contains(alphabet[c]) { continue }
                
                index -= 1
                s[i] = alphabet[c]
            }
        }
    }
    return s.joined()
}

문제를 받고 생각한 것.

1. 현재 알파벳에서 index만큼 뒤에 있는 알파벳을 알기 위해선, 현재 알파벳의 index를 알아야 한다.
2. 현재 알파벳의 index에서 index만큼 뒤에 있는 알파벳을 찾을 때, 알파벳의 최대 갯수(26개)를 넘기면 처음으로 돌아와야 한다.
3. skip에 포함되면 넘긴다
4. 포함되지 않으면 index(count 용도)에서 -1을 해주고,
5. s[i]에 현재 인덱스의 알파벳을 반영한다.
6. 위의 과정을 반복한다.