[프로그래머스] 시저 암호

<문제>

---

<전략짜기>

일단 알파벳 대소문자는 ASCII코드를 가지고 있어서, s를 대/소문자 범위에 따라 체크하는 것도 하나의 방법일 듯 한데.. (나는 ASCII코드를 사실 잘 모름 ㅋ) 알파벳 순서는 어차피 알고 있는 거니까 차라리 알파벳을 미리 정의해놓고, s의 문자들을 체크하면서 정의해둔 알파벳 문자에서 n만큼 뒤에 있는 문자로 대체하는 걸로 풀어야겠다. z를 넘어가면 Out of Index니까.. 원형배열처럼 % 연산자를 사용해서 Index를 찾아야겠다.

 

<코드>

let alpha = "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
let lowerAlpha = alpha.lowercased()
let arrayAlpha = Array(alpha)
let arrayLowerAlpha = Array(lowerAlpha)

func solution(_ s:String, _ n:Int) -> String {
    var array = Array(s)
    for (i, x) in array.enumerated() {
        if alpha.contains(x) {
             array[i] = arrayAlpha[(Int(arrayAlpha.firstIndex(of: x)!)+n) % alpha.count]
        } else if lowerAlpha.contains(x) {
            array[i] = arrayLowerAlpha[(Int(arrayLowerAlpha.firstIndex(of: x)!)+n) % alpha.count]
        }
    }
    return String(array)
}

 

<Lesson Learned>

일단 map으로도 충분히 가능할 것 같은데 익숙치 않아서 일단 배열로 문제 푼 상황.. ㅜ.ㅜ 그래서 그런지 코드가 굉장히 깔끔하지 않은 듯(?) 하다 ㅋㅋ map으로 바꿔도 로직은 똑같이 들어가긴 할텐데..보기에 깔끔한(?) ㅎㅎ 그리고 이문제는 시간제한 or 메모리제한 에 따라 약간 로직을 바꿀 수 있을 것 같다. 아마 2가지는 시간/메모리가 Trade Off 인.. 😭 그래도 이 문제는 s의 길이가 최대 8000 이니까, 차라리 lowerAlpha를 따로 정의해놓고 위의 풀이처럼 푸는게 나을 것 같네 🤟🤟

let alpha = "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
let arrayAlpha = Array(alpha)

func solution(_ s:String, _ n:Int) -> String {
    var originArray = Array(s)
    var array = Array(s)
    for (i, x) in array.enumerated() {
        if alpha.contains(x.uppercased()) {
            array[i] = arrayAlpha[(Int(arrayAlpha.firstIndex(of: Character(x.uppercased()))!)+n) % alpha.count]
        }
    }
    for (i, x) in originArray.enumerated() {
        if x.isLowercase {
            array[i] = Character(array[i].lowercased())
        }
    }
    return String(array)
}

 

 

 

출저 : 프로그래머스 시저 암호