[BOJ] 5598번 카이사르 암호

카이사르 암호

• 문자열을 이용한 구현문제 였습니다. 입력된 문자열에 3개씩 건너뛴 문자가 있는데 이를 원래 문자열로 복구하는 문제입니다.
• 입력된 문자열을 for문을 통해 순회하며 예외처리!! A,B,C같은 경우는 변환후 값을 보시면 X,Y,Z 로 되어있기 때문에 별도 if문으로 append로 추가하였습니다. 그외에 값들은 array에 입력된 문자의 index를 찾은후 - 3을 하여 append하였습니다!!

var array:[String.Element] = ["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"]
var S = readLine()!
var result:String = ""


if  S.count > 1 || S.count <= 1000{
    S.forEach { t in
        if String(t).contains("A") {
            result.append("X")
        } else if String(t).contains("B") {
            result.append("Y")
        } else if String(t).contains("C") {
            result.append("Z")
        } else if array.contains(t) {
            result.append(array[array.firstIndex(of: t)! - 3])
        }
    }
    print(result)
}