Programming/프로그래머스

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

pental 2020. 10. 11. 18:37

def solution(s, n):
    answer = []
    for i in s:
        if i.isalpha():
            if( 65 <= ord(i) <= 90):
                if ord(i) + n > 90:
                    result = ord(i) + n - 26
                    answer.append(chr(result))
                else :
                    result = ord(i) + n
                    answer.append(chr(result))
            else :
                if ord(i) + n > 122:
                    result = ord(i) + n - 26
                    answer.append(chr(result))
                else :
                    result = ord(i) + n
                    answer.append(chr(result))
        else:
            answer.append(' ')
    print(answer)
    return ''.join(answer)