[파이썬/알고리즘] 프로그래머스 - 시저 암호

[문제]

어떤 문장의 각 알파벳을 일정한 거리만큼 밀어서 다른 알파벳으로 바꾸는 암호화 방식을 시저 암호라고 합니다. 예를 들어 "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)):
        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)

 

isupper() 과 islower()는 대문자인지, 소문자인지를 반환한다.

즉 isupper()를 썼는데 문자가 대문자로 되어있다면 True, 소문자로 되어있다면 False를 반환한다.

만약 세 글자 중 두 글자가 대문자인데 한 글자가 소문자로 되어있어도 False를 반환한다.

 

s = list(s)

문자열로 들어온 s를 각 글자마다 변환을 해줘야해서 리스트로 변환해준다.

 

for i in range(len(s)):

리스트로 변환 한 이유와 마찬가지로 각 글자마다 변환을 해줘야 하기 때문에 길이만큼 반복문을 돌게한다.

 

if s[i].isupper():
     s[i] = chr((ord(s[i])-ord('A')+ n)%26+ord('A'))

만약 첫 번째 글자가 대문자라면 두 번째 줄 코드를 실행하며 ord(s[i])는 i 번째 알파벳의 ASCII를 반환한다.

그러고나서 ord('A)를 빼준다. 빼주는 이유는, A와 B는  ASCII로 변환하면 97, 98이다. 따라서 ord('A) - ord('A')를 하면 0이 되고 ord('B') - ord('A)를 하면 1이 나온다. 그리고 여기서 + n을 더하고 26으로 나눈 나머지를 구해준다.

 

26으로 나눈 나머지를 구하지 않게 되면, n이 27일 경우 알파벳 A(0) ~ Z(26) 범위를 벗어나게 되고 알파벳 안에서 계산하면 n = 27과 n = 1이 서로 같기 때문에 26으로 나눈 나머지를 구한다. 그리고 ord('A)를 더해줌으로써 다시 ASCII로 만들고

chr을 통해서 문자열로 반환하여 알파벳이 나온다.

 

elif s[i].islower():
      s[i] = chr((ord(s[i]) - ord('a') + n)%26 + ord('a'))

소문자인 경우에도 동일한 로직을 거치며, 설명은 위 대문자와 동일하다.
    

return "".join(s)

각 리스트를 공백없이 문자열로 반환한다.