[Python] Lv1.시저 암호

https://programmers.co.kr/learn/courses/30/lessons/12926

코딩테스트 연습 - 시저 암호

 

 

 

문제 설명

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

 

제한 조건

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

 

나의 접근 방법 및 풀이

주어진 문자열에서 하나하나 체크해야되기때문에 for문을 이용해야겠다고 생각했다. 그 다음 n만큼 증가한 문자는 어떻게 만들어야 하지 하다가 아스키 코드가 생각났다.

처음에는 'z', 'Z', 그 외 문자, 공백이 경우를 나누어 코드를 짰지만, 'B', 'b' 이상의 문자에 25을 더하면 해당 범위를 넘어가는 문제가 발생했다.

그래서 문자가 공백인 조건, 대문자인 조건, 소문자인 조건으로 다시 조건을 나누었다. 공백인 경우에는 결과를 return할 문자열에 공백을 더하면 된다. 그렇지 않은 경우인 두경우에는 아래처럼 풀어나간다.

먼저, 문자의 대소문자를 구분한다.

 

<대문자인 경우>

이때, ord()를 이용해 문자 → 아스키 코드 로 변한값과 n을 더한 값이 90을 넘어가면 'A' ~ 'Z' 사이크을 넘어가기 때문에 -26을 해주어야 한다. 그리고 chr()을 아스키코드→ 문자이용하여 변환하여 answer에 더한다. 넘어가지 않는다면, 해당 문자의 아스키코드와 n을 더한 값을 chr()로 변환하면 된다.

 

<소문자인 경우>

소문자인 경우는 대문자인 경우와 비슷한 방법으로 진행되면, 이때 조건값을 122로 설정한다.

def solution(s, n):
    answer = ''
    for i in s:
        if i != ' ':
            if 'A' <= i <= 'Z':
                num = ord(i)+n
                if num > 90:
                    answer += chr(num-26)
                else:
                    answer += chr(num)
            elif 'a' <= i <= 'z':
                num = ord(i)+n
                if num > 122:
                    answer += chr(num-26)
                else:
                    answer += chr(num)
        else:
            answer +=' '
    return answer

solution("AaZz", 25);
solution("a    b", 1);
solution("a b ", 1);

 

 

다른 풀이

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)
    # 주어진 문장을 암호화하여 반환하세요.


# 실행을 위한 테스트코드입니다.
print('s는 "a B z", n은 4인 경우: ' + caesar("a B z", 4))

대소문자를 isupper()와 islower()로 구분하고,

ord(s[i])-ord('A')+n은 (현재 문자의 아스키코드값 + n) -25 값이며 이를 26으로 나눈 나머지와 'A'의 아스키코드값을 더하여 다시 문자열로 나누었다. 소문자도 이와 같이 진행하였고 최종적으로는 리스트를 문자열로 바꾸기 위해 join()을 이용하여 return 하였다.