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

프로그래머스

---

문제풀이를 직관적으로 이해할 수 있도록 고차함수를 사용하지 않은 풀이와
고차함수를 사용한 풀이로 나누어 공부하고 있습니다.
- 문제에 따라 하나의 풀이만 있을 수 있습니다.

 

고차함수를 사용하지 않은 풀이

func solution(_ s:String, _ n:Int) -> String {
let upper = Array("abcdefghijklmnopqrstuvwxyz".uppercased())
let lower = Array("abcdefghijklmnopqrstuvwxyz")
var result = ""

for i in s {
    if i == " " { result.append(" ")}
    else {
        if upper.contains(i) {
            result.append(String(upper[(upper.firstIndex(of: i)! + n) % 26]))
        } else {
            result.append(String(lower[(lower.firstIndex(of: i)! + n) % 26]))
        }
    }
}
    return result
}

•  

 

고차함수를 사용한 풀이

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

• 문자열을 utf8 형식으로 고차함수를 돌리면 i 값들은 String.UTF8View.Element 라는 타입을 가지고 나온다.
• utf8의 표현은 아스키코드와 같다.
• 따라서 Int로 초기화 해준 뒤 알파벳의 아스키코드 범위를 비교해 대문자인지 소문자인지 switch문을 통해 비교할 수 있다.
• UnicodeScalar(num)!는 Int형식의 아스키코드를 넣으면 해당 문자를 반환해준다.

 

고차함수를 사용한 풀이

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
    })
}

•