백준: 11655 ROT13

문제

11655번: ROT13

 

문제 풀이

 ROT13은 문자가 입력되면 소문자, 대문자를 구분하여 각 문자로 부터 13번째 이후의 알파벳으로 변경하는 암호화 방식이다. 해당 암호화 방식을 구현하면 문제를 해결 할 수 있다.

 

• 소문자와 대문자의 인덱스를 찾기 위한 딕셔너리를 초기화 한다.
  • stirng 상수를 활용하면 좋다.
• 대문자는 ASCII코드 65 부터, 소문자는 97 부터 시작하므로 이를 활용하여 +13된 값을 찾는다.

 

코드

import string

LOWER = {c: idx  for idx, c in enumerate(string.ascii_lowercase)}
UPPER = {c: idx for idx, c in enumerate(string.ascii_uppercase)}
IDX, ALPHA = 0, 1


def encrypt(c, lower):
    lst = UPPER
    factor = 65
    if lower:
        factor = 97
        lst = LOWER
    return chr(factor + ((lst[c] + 13) % 26))


def rot13(source):
    ret = ""
    for c in source:
        if c.isupper():
            ret += encrypt(c, False)
        elif c.islower():
            ret += encrypt(c, True)
        else:
            ret += c
    return ret

if __name__ == '__main__':
    print(rot13(input()))

 코드를 작성하고 제출 후에 생각해보니 딕셔너리를 만들 때 인덱스가 아닌 아스키로 초기화 했으면 더 깔끔할 거 같다. 숏코딩의 바이트 수가 51B가 있어서 이건 어떻게 짠거지...? 하고 들어가 봤는데 built-in function 중에 rot13이 구현되어 있었다...

 

 

codecs로 구현한 풀이

import codecs
print(codecs.encode(input(),"rot13"))