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

프로그래머스 LV.1 모음

시저 암호

문제 설명

어떤 문장의 각 알파벳을 일정한 거리만큼 밀어서 다른 알파벳으로 바꾸는 암호화 방식을 시저 암호라고 합니다. 예를 들어 "AB"는 1만큼 밀면 "BC"가 되고, 3만큼 밀면 "DE"가 됩니다. "z"는 1만큼 밀면 "a"가 됩니다. 문자열 s와 거리 n을 입력받아 s를 n만큼 민 암호문을 만드는 함수, solution을 완성해 보세요.

제한 사항

• 공백은 아무리 밀어도 공백입니다.
• s는 알파벳 소문자, 대문자, 공백으로만 이루어져 있습니다.
• s의 길이는 8000 이하입니다.
• n은 1 이상, 25 이하인 자연수입니다.

입출력 예

s	n	result
"AB"	1	"BC"
"z"	1	"a"
"a B z"	4	"e F d"

---

제출

func solution(_ s:String, _ n:Int) -> String {
    return s.utf8.map {
        var code = Int($0)
        switch code {
            case 65...90:
                code = (code + n - 65) % 26 + 65
            case 97...122:
                code = (code + n - 97) % 26 + 97
            default:
                break
        }
        return String(UnicodeScalar(code)!)
    }.joined()
}

입력받은 문자열을 utf8 프로퍼티로 접근한다.
문자열의 코드값이 65부터 90 사이(대문자 알파벳) 이거나 97부터 122 사이(소문자 알파벳) 라면 n 만큼 증가시킨다.
이후 코드를 문자열로 변환해 반환한다.

다른 풀이

func solution(_ s:String, _ n:Int) -> String {
    let alphabets = "abcdefghijklmnopqrstuvwxyz".map { $0 }
    return String(s.map {
        guard let index = alphabets.firstIndex(of: Character($0.lowercased())) else { return $0 }
        let letter = alphabets[(index + n) % alphabets.count]
        return $0.isUppercase ? Character(letter.uppercased()) : letter
    })
}

소문자 알파벳을 배열로 만든다.
문자열의 각 문자의 인덱스를 구한 후, 인덱스+n의 값을 구해 letter변수에 저장한다.
문자열의 문자가 대문자라면, letter 변수를 대문자로 변환하여 반환한다.