(Python/LV1) 시저 암호

문제 출처: https://school.programmers.co.kr/learn/courses/30/lessons/12926

프로그래머스

풀이:

def solution(s, n):
    s = list(s)
    for i in range(len(s)):
        print(s[i])
        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)
print(solution("AB", 2 ))