프로그래머스 연습문제: 시저 암호 (python)

문제

 

풀이

def solution(s, n):
    answer = ""
    for i in s:
        if 97<=ord(i)<=122:
            if ord(i)+n > 122:
                answer+=chr(ord(i)+n - 26)
            else:
                answer+=chr(ord(i)+n)
        elif 65<=ord(i)<=90:
            if ord(i)+n > 90:
                answer+=chr(ord(i)+n - 26)
            else:
                answer+=chr(ord(i)+n)
        elif i==" ":
            answer += " "

    return answer

 

ord와 chr 함수를 이용해서 풀어주었다.

ascii에서 97~122까지는 a~z까지 65부터 90까지는 A-Z인것을 이용해서 범위를 나눠주어서 풀었다.

그리고 공백값이 들어올 수 있으니 공백에 대한 예외처리도 꼭 해줘야한다.