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

문제

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

제한 조건

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

나의 풀이:

def solution(s, n):
    answer = ''
    cap = "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
    low = "abcdefghijklmnopqrstuvwxyz"
    
    for i in s:
        if i in cap:
            i = cap[(cap.index(i)+n)%26]
            answer += i
        elif i in low:         
            i = low[(low.index(i)+n)%26]
            answer += i
        else:
            answer += ' '
    return answer

ASCII 코드를 사용해서 풀 수도 있을 것 같으나 지금 당장 이 문제를 시험장에서 받는다고 하면 ASCII 코드 변환 함수를 내가 모르기 때문에 그냥 배열을 사용하여 문제를 풀었다.

다른 사람들의 풀이:

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)

ASCII 코드 및 캐릭터 대소문자 구분 함수를 사용한 풀이다.

• ord(): 문자를 10진수 ASCII 코드로 변환
• chr(): 숫자를 10진수 ASCII 코드로 인식하여 문자로 변환
• isupper(): 문자가 대문자면 True 반환
• islower(): 문자가 소문자면 True 반환