백준 11655번: ROT13 (python, 파이썬)

• 문제

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으로 암호화한 내용을 출력한다.

---

• 코드

import sys

num2str = {} # 숫자 -> 문자
str2num = {} # 문자 -> 숫자
count = 1
for i in list('abcdefghijklmnopqrstuvwxyz'):
    str2num[i] = count
    num2str[count] = i
    count += 1

# 최종 답
result = ''

# 마지막 \n까지 인식하기 때문에 마지막 전까지만 for문
for i in list(sys.stdin.readline())[:-1]:
    # 공백 or 숫자
    if i == ' ' or i.isdigit():
        result += i
        continue
    
    # 뒤에서 대/소문자 check해야해서 다른 변수에 저장
    i_low = i.lower()
    
    # 알파벳 z를 넘어가는 경우 다시 처음으로 돌리는 logic
    value = str2num[i_low] + 13
    if value > 26:
        value -= 26
    
    # 소문자일때
    if i.islower():
        result += num2str[value]
    # 대문자일때
    else:
        result += num2str[value].upper()

print(result)

---

• 출처

www.acmicpc.net/problem/11655