문제 설명

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

제한 조건

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

입출력 예

s	n	result
"AB"	1	"BC"
"z"	1	"a"
"a B z"	4	"e F d"

접근 방법

이번 문제는 파이썬의 ord()함수와 chr()함수를 이용하여 대문자와 소문자를 나눠서 처리하는 방식으로 해결하였다. 대문자는 Z를 넘어가면 다시 A부터 시작할 수 있도록 처리하였고, 소문자는 z를 넘어가면 다시 a부터 시작할 수 있도록 처리하였다. 대소문자 비교는 isupper(), islower()를 통해 비교하였다.

• s를 리스트로 변환한다. (문자열은 불변 객체이기 때문에 가변 객체로 바꿔준다.)
• s의 길이만큼 반복하는 i에 대한 for문을 돌린다.
  -> 만약 s[i]가 대문자일 경우, s[i]를 s[i]의 아스키코드값 - A의 아스키코드값 + n을 26으로 나눈 나머지에 A의 아스키코드값을 더한 값을 문자로 변환한다.
  -> 만약 s[i]가 소문자일 경우, s[i]를 s[i]의 아스키코드값 - a의 아스키코드값 + n을 26으로 나눈 나머지에 a의 아스키코드값을 더한 값을 문자로 변환한다.
• answer에 s를 문자열로 변환한 값을 넣어준다.
• answer를 반환한다.

solution.py

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'))
    answer=''.join(s)
    return answer