[ 프로그래머스 / PYTHON ] 시저 암호

문제설명

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

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

내가 생각한 방식 - 알고리즘

1. 문자열로 받은 s를 아스키 코드로 변환한다
2. 빈 문자열이 아니고 대문자일 때(아스키 코드상 65~90) n만큼 더한다.(=n만큼 민다)
3. 만약 대문자 아스키 번호(Z = 90)를 넘는다면(91부터) 65(A의 아스키코드)에 넘는 수 만큼 더해준다.(A에서부터 민다.)
4. 소문자에서도 2~3번 동일
5. 아스키 코드를 다시 문자열로 변환 시켜 반환한다.

코드

def solution(s, n):
    li = list(s)
    for i in range(len(li)):
        if li[i] != " ":
            li[i] = ord(li[i])

    for i in range(len(li)):
        if li[i] != " " and 65 <= li[i] <= 90:
            li[i] += n
            if li[i] > 90:
                temp = li[i] - 91
                li[i] = 65 + temp
        elif li[i] != " " and 97 <= li[i] <= 122:
            li[i] += n
            if li[i] > 122:
                temp = li[i] - 123
                li[i] = 97 + temp
            
    for i in range(len(li)):
        if li[i] != " ":
            li[i] = chr(li[i])
            
    return ''.join(li)