[프로그래머스/Programmers] 시저 암호 - Python

Link

코딩테스트 연습 - 시저 암호

 

Solution

def solution(s, n):
    answer = ''

    for i in s:
        if i == ' ':
            answer += ' '
            continue

        r = ord(i)+n

        if r > 122 and 97 <= ord(i) and ord(i) <= 122:
            r -= 26
        elif r > 90 and 65 <= ord(i) and ord(i) <= 90:
            r -= 26

        answer += chr(r)

    return answer

🔑 key point: ord(), chr()

 

Other Solution

def solution(s, n):
    s = list(s)

    for i in range(len(s)):
        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)