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

문제 설명

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

 

제한 조건

• 공백은 아무리 밀어도 공백입니다.
• s는 알파벳 소문자, 대문자, 공백으로만 이루어져 있습니다.
• s의 길이는 8000이하입니다.
• n은 1 이상, 25이하인 자연수입니다.

 

def solution(s, n):
    s = list(s) # 리스트로 변환
    for i in range(len(s)):
        # 아스키코드 a ~ z 26개
        # 소문자일 때 
        if s[i].islower():
            s[i] = chr((ord(s[i]) - ord('a') + n) % 26 + ord('a'))
        # 대문자일 때
        elif s[i].isupper():
            s[i] = chr((ord(s[i]) - ord('A') + n) % 26 + ord('A'))
    return "".join(s)

chr( ) 함수 : 숫자를 인자로 주어 아스키 코드에 해당하는 문자 반환(아스키코드 -> 문자)
ord( ) 함수 : 문자의 아스키 코드를 반환(문자 -> 아스키코드)

 

알고리즘

1. 특정 알파벳에서 맨 처음 알파벳 A 또는 a를 빼고 그로부터 n만큼 밀어주기 때문에 n을 더한 후 총 알파벳의 개수 26으로 나눈 나머지 -> 알파벳의 자릿수(a로부터 얼마나 떨어져 있는지)

2. 첫 알파벳(A or a)의 아스키코드로부터 위에서 얻은 자릿수를 더함으로써 그 값을 구함