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

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

 

def solution(s, n):
    answer = []
    tmp = ''
    letter_upper = list("ABCDEFGHIJKLMNOPQRSTUVWXYZABCDEFGHIJKLMNOPQRSTUVWXYZ")
    letter_lower = list("abcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwxyz")
    
    for i in s:

        for k in range(0,26):
            if i == letter_upper[k]:
                tmp = str(letter_upper[k+n])
                answer.append(tmp)

            elif i == letter_lower[k]:
                tmp = str(letter_lower[k+n])
                answer.append(tmp)
                
        if i == " ":
            tmp = " "
            answer.append(tmp)

    answer = ''.join(answer)

    return answer