import Foundation

let alphabet: [String] = ["A", "B", "C", "D", "E", "F", "G", "H", "I", "J", "K", "L", "M", "N", "O", "P", "Q", "R", "S", "T", "U", "V", "W", "X", "Y", "Z"]

func solution(_ s:String, _ n:Int) -> String {
    return s.map { s in
        if s.description == " " {
            return " "
        }else {
            guard let ascii = s.asciiValue else {
                return ""
            }
            
            var isLowercased = false
            var index = Int(ascii - 65.uValue)
            if ascii >= 97.uValue && ascii <= 122.uValue {
                isLowercased = true
                index = Int(ascii - 97.uValue)
            }
            
            index = (index + n) % (alphabet.count)
            if isLowercased {
                return alphabet[index].lowercased()
            }
            
            return alphabet[index]
        }

    }.joined()
}

extension Int {
    var uValue: UInt8 {
        return UInt8(self)
    }
}