백준 4446 ROT13 (파이썬)

https://www.acmicpc.net/problem/4446

4446번: ROT13

 

입력 문자가 "여러 줄" 인 점을 고려해야합니다.

try catch문으로 여러 줄을 받게 만들면 나머지는 쉬운 구현문제입니다.

 

           

예시             1 2 3 4 5 6

                     ^      ^

                     end   st      

 

부연 설명 : 길이 6인 배열의 4번 인덱스에서 -3만큼 이동 > 1번 인덱스로 이동

                            4 - 3 = 1

                            (4 + 3) % 6 = 1

           

 

a = ['a', 'i', 'y', 'e', 'o', 'u']
b = ['b', 'k', 'x', 'z', 'n', 'h', 'd', 'c', 'w', 'g', 'p', 'v', 'j', 'q', 't', 's', 'r', 'l', 'm', 'f']
while True:
    try:
        S = input()
        ans = ''
        for i in range(len(S)):
            if 65 <= ord(S[i]) <= 90:
                tmp = S[i].lower()
                if tmp in a:
                    idx = (a.index(tmp) + 3) % len(a)
                    ans += a[idx].upper()
                else:
                    idx = (b.index(tmp) + 10) % len(b)
                    ans += b[idx].upper()
            elif 97 <= ord(S[i]) <= 122:
                if S[i] in a:
                    idx = (a.index(S[i]) + 3) % len(a)
                    ans += a[idx]
                else:
                    idx = (b.index(S[i]) + 10) % len(b)
                    ans += b[idx]
            else:
                ans += S[i]
        print(ans)
    except:
        break