[Algorithm] 프로그래머스 시저 암호 Level 1 (Swift)

코딩테스트 연습 - 시저 암호

문제 풀이

• 아스키코드 이용해서 풀음
• 입력 받은 문자열을 배열에 저장한 후, 배열을 하나하나 for 문을 통해 순회
  • 아스키코드를 구한 후, 아스키코드에 n을 더해주었다.
• 공백은 무시하고, 대문자인 경우와 소문자인 경우로 나눠서 풀음
  • 대문자인 경우, 아스키코드에 n을 더한 결과가 90을 넘을 경우 26을 뺐음.
  • 소문자인 경우, 아스키코드에 n을 더한 결과가 122를 넘을 경우 26을 뺐음.
  • 그리고 나서 변경된 아스키코드를 Character형으로 변경해서 배열에 저장.
• for 문을 다 돌고 나면, 배열을 String으로 변환하여 리턴.

테스트 케이스

print(solution("AB", 1)) // BC
print(solution("z", 1)) // a
print(solution("a B z", 4)) // e F d
print(solution("bC", 25)) // aB
print(solution("abcdefghijklmnopqrstuvwxyz", 2)) // cdefghijklmnopqrstuvwxyzab

소스 코드 (Swift)

//
//  시저암호.swift
//  Programmers
//
//  Created by 안상희 on 2022/01/30.
//

import Foundation

func solution(_ s:String, _ n:Int) -> String {
    var str = Array(s)
    for i in 0..<str.count {
        if str[i] == " " { continue } // 공백 무시
        
        var asciiCode = Int(str[i].asciiValue!)
        asciiCode += n
        
        if str[i].isUppercase { // 대문자
            if asciiCode > 90 {
                asciiCode -= 26
            }
        } else { // 소문자
            if asciiCode > 122 {
                asciiCode -= 26
            }
        }

        str[i] = Character(UnicodeScalar(asciiCode)!)
    }
    return String(str)
}