[백준] 11655번 ROT13 (Python3)

문제

ROT13은 카이사르 암호의 일종으로 영어 알파벳을 13글자씩 밀어서 만든다.

예를 들어, "Baekjoon Online Judge"를 ROT13으로 암호화하면 "Onrxwbba Bayvar Whqtr"가 된다. ROT13으로 암호화한 내용을 원래 내용으로 바꾸려면 암호화한 문자열을 다시 ROT13하면 된다. 앞에서 암호화한 문자열 "Onrxwbba Bayvar Whqtr"에 다시 ROT13을 적용하면 "Baekjoon Online Judge"가 된다.

ROT13은 알파벳 대문자와 소문자에만 적용할 수 있다. 알파벳이 아닌 글자는 원래 글자 그대로 남아 있어야 한다. 예를 들어, "One is 1"을 ROT13으로 암호화하면 "Bar vf 1"이 된다.

문자열이 주어졌을 때, "ROT13"으로 암호화한 다음 출력하는 프로그램을 작성하시오.

입력

첫째 줄에 알파벳 대문자, 소문자, 공백, 숫자로만 이루어진 문자열 S가 주어진다. S의 길이는 100을 넘지 않는다.

출력

첫째 줄에 S를 ROT13으로 암호화한 내용을 출력한다.

나의 첫 번째 풀이

문자열의 각각의 문자에서 아스키코드 값으로 13을 더해줘서 풀고자 했다.
하지만 13을 더해주었을 때 z의 값을 넘어가는 알파벳(n~z)은 다시 맨 처음 알파벳으로 돌아오게 만들어줘야 했다.
그래서 13을 더한 값에서 z 값을 나누어준 값을 a 값에 더해서 돌아오도록 만들어 주었다.

ord('a') + (ord(입력값)+13) % ord('z') -1

ex)

N의 ROT13값 = A

'N'의 아스키 코드 : 78 (그냥 13을 더한 아스키코드는 91로 'Z'의 아스키 코드인 90을 초과)

65 + (78+13)%90 -1 = 65

'A' 의 아스키 코드 : 65
'Z' 의 아스키 코드 : 90

'a' 의 아스키 코드 : 97
'z' 의 아스키 코드 : 122

import sys

st = sys.stdin.readline()
answer = ''

for i in st:
    ROT = ord(i)+13
    if 'a' <= i <= 'z':
        if ROT > ord('z'):
            answer += chr(ord('a') + (ROT)%ord('z') -1)
        else:
            answer += chr(ROT)

    elif 'A' <= i <= 'Z':
        if ROT > ord('Z'):
            answer += chr(ord('A') + (ROT)%ord('Z') -1)
        else:
            answer += chr(ROT)
    else:
        answer += i

print(answer)