[파이썬] 시저 암호

문제 설명

어떤 문장의 각 알파벳을 일정한 거리만큼 밀어서 다른 알파벳으로 바꾸는 암호화 방식을 시저 암호라고 합니다. 예를 들어 AB는 1만큼 밀면 BC가 되고, 3만큼 밀면 DE가 됩니다. z는 1만큼 밀면 a가 됩니다. 문자열 s와 거리 n을 입력받아 s를 n만큼 민 암호문을 만드는 함수, solution을 완성해 보세요.

제한 조건

공백은 아무리 밀어도 공백입니다.
s는 알파벳 소문자, 대문자, 공백으로만 이루어져 있습니다.
s의 길이는 8000이하입니다.
n은 1 이상, 25이하인 자연수입니다.

입출력 예시

s	n	result
AB	1	BC
z	1	a
a B z	4	e F d

나의 코드

def solution(s, n):
    answer = ""
    alpha = "AaBbCcDdEeFfGgHhIiJjKkLlMmNnOoPpQqRrSsTtUuVvWwXxYyZz"

    for i in range(len(s)):
        for j in range(len(alpha)):
            if(s[i] == alpha[j]):
                answer += alpha[int((j+2*n)%len(alpha))]
            elif(s[i] not in alpha):
                answer += " "
                break
               
    return answer

나는 먼저 알파벳 대문자와 소문자를 한 변수에 모두 넣어놓았다. 문자열 s 가 들어오면 변수 alpha 를 돌면서 s 의 글자와 alpha 의 글자를 하나씩 비교한다. 그러다가 alpha 에서 같은 글자가 나오면 int((j + 2*n) % len(alpha)) 식을 이용하여 n 칸 뒤로 이동한다. 만약 주어진 알파벳이 z인 경우 한칸 더 가면 a로 돌아와야하기 때문에 circular queue 에서 쓰는 인덱스 방식을 사용했다.
만약 입출력 예시에서 "a B z" 처럼 글자 중간에 공백이 있는 경우를 대비하여 elif 문을 작성했는데, 공백이 나오는 경우에는 answer 에다가 공백 하나를 더해주었다.

다른 사람의 코드

def solution(s, n):
    s = list(s)
    for i in range(len(s)):
        if s[i].isupper():
            s[i]=chr((ord(s[i])-ord('A')+ n)%26+ord('A'))
        elif s[i].islower():
            s[i]=chr((ord(s[i])-ord('a')+ n)%26+ord('a'))

    return "".join(s)

다른 사람의 풀이에서는 나처럼 알파벳을 미리 정의해놓지 않고 ord 함수를 사용하여 알파벳의 ASCII 코드를 조정했다.