[test] 시저 암호 - python

Q.

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

My A.

def solution(s, n):
    alphabet = 'abcdefghijklmnopqrstuvwxyz'
    u_alpha = alphabet.upper()
    answer = ''
    for letter in s:
        if letter.isspace() == True:
            answer = answer + " "
        elif letter.islower() == True:
            position = alphabet.find(letter)
            newPosition = (position+n) % 26
            answer = answer + alphabet[newPosition]
        else:
            position = u_alpha.find(letter)
            newPosition = (position+n) % 26
            answer = answer + u_alpha[newPosition]
    return answer

1. alphabet에 알파벳 배열 넣어주기.
2. 대문자 전용 배열도 만들어주기
3. answer 은 빈 문자열.
4. s의 각 레터를 반복할 때 만약 공백이면 answer에 공백 추가해주기 - isspace()활용
5. 그 외의 경우, 소문자라면 position에 원래 인덱스값을 찾아 할당하고, 그것에 밀 수인 n을 더한다음 만약 알파벳 갯수인 26보다 클 경우를 대비해서 26으로 나눈 나머지를 newP에 할당.
6. answer에 알파벳배열에서 새로운인덱스값(newP)에 해당하는 알파벳 붙여주기.
7. 대문자면 대문자 배열에서 찾아 붙이기.
8. 최종 answer 값 리턴하기.

++ if절의 경우 조건문 == True를 구지 넣지 않아도 됨.
if letter.isspace() : 이렇게

참고

파이썬 문자열 메서드 정리

other solution

def caesar(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)

chr(i)
유니코드 코드 포인트가 정수 i 인 문자를 나타내는 문자열을 돌려줍니다. 예를 들어, chr(97) 은 문자열 'a' 를 돌려주고, chr(8364) 는 문자열 '€' 를 돌려줍니다. 이 것은 ord() 의 반대입니다.

ord(c)
하나의 유니코드 문자를 나타내는 문자열이 주어지면 해당 문자의 유니코드 코드 포인트를 나타내는 정수를 돌려줍니다. 예를 들어, ord('a') 는 정수 97 을 반환하고 ord('€') (유로 기호)는 8364 를 반환합니다. 이것은 chr() 의 반대입니다.

join() - 리스트의 문자열들을 합치는 역할

"".join(리스트)

활용함.