시저 암호

[문제]

어떤 문장의 각 알파벳을 일정한 거리만큼 밀어서 다른 알파벳으로 바꾸는 암호화 방식을 시저 암호라고 합니다. 예를 들어 "AB"는 1만큼 밀면 "BC"가 되고, 3만큼 밀면 "DE"가 됩니다. "z"는 1만큼 밀면 "a"가 됩니다. 문자열 s와 거리 n을 입력받아 s를 n만큼 민 암호문을 만드는 함수, solution을 완성해 보세요.

 

 

[풀이]

아스키 코드를 이용해서 문제를 풀면 된다.

 

A~Z 는 아스키 코드로 65 ~ 90

a ~ z 는 아스키 코드로 97 ~ 122 

 

ord 함수로 문자를 아스키 코드로 변환할 수 있으며,

chr 함수로 아스키 코드를 문자로 변환할 수 있다.

 

문자가 문자를 나타내는 아스키 코드 범위 밖으로 나가지 않도록 작업을 해줘야 한다. (랩어라운드라고 한다)

(ord(문자) - ord('a 또는 A') + n) % 26 (알파벳 개수) + ord('a 또는 A')

현재 문자가 기준(a 또는 A) 문자열로 부터 얼마나 떨어져 있는지 구하고 n을 더해준 뒤 해당 값에 모듈연산을 해서 범위가 벗어나지 않도록 해준다. 마지막으로 기준 문자열을 더해줘서 정확한 위치를 구해준다.

 

def solution(s, n):
    
    return ''.join(
        chr((ord(char) - ord('A') + n) % 26 + ord('A')) 
        if char.isupper() 
        else chr((ord(char) - ord('a') + n) % 26 + ord('a'))
        if char.islower()
        else char
        for char in s
    )