프로그래머스 시저 암호
in Dev / Codingtest on CodingTest
프로그래머스 코딩테스트 시저 암호
문제
어떤 문장의 각 알파벳을 일정한 거리만큼 밀어서 다른 알파벳으로 바꾸는 암호화 방식을 시저 암호라고 합니다 예를 들어 “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" |
풀이
def solution(s, n):
answer = []
for char in s:
if char.isalpha():
start = ord('A') if char.isupper() else ord('a')
answer.append(chr((ord(char) - start + n) % 26 + start))
else:
answer.append(char)
return ''.join(answer)
- 먼저 빈 리스트를 생성하여 암호화된 문자들을 저장한다.
- 문자열
s의 각 문자char에 대해 반복한다. if char.isalpha()이 조건문은char가 알파벳 문자인지 확인한다.isalpha()함수는 문자가 알파벳이면True를, 아니면False를 반환한다.
start = ord('A') if char.isupper() else ord('a')이 부분은char가 대문자인지 소문자인지에 따라 시작점을 정한다.isupper()함수는 문자가 대문자이면True를, 아니면False를 반환한다.ord('A')는 대문자 ‘A’의 아스키 코드 값을 반환한다.ord('a')는 소문자 ‘a’의 아스키 코드 값을 반환한다.
answer.append(chr((ord(char) - start + n) % 26 + start))이 부분은char를n만큼 시저 암호 방식으로 암호화한다.ord(char)는char의 아스키 코드 값을 반환한다.ord(char) - start는char를 0부터 시작하는 알파벳 인덱스로 변환한다.(ord(char) - start + n) % 26는char를n만큼 오른쪽으로 밀어서 암호화하고, 26으로 나눈 나머지를 구한다.chr((ord(char) - start + n) % 26 + start)는 암호화된 인덱스를 다시 문자로 변환한다.
else:이 부분은char가 알파벳이 아닌 경우이다.- 알파벳이 아닌 문자는 암호화하지 않고 그대로
answer에 추가한다.
- 알파벳이 아닌 문자는 암호화하지 않고 그대로
- 마지막으로
''.join(answer)는answer리스트의 문자들을 연결하여 문자열로 변환한다.