[프로그래머스] level1 - 시저 암호 (Python)

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이하인 자연수입니다.

입출력 예

s	n	result
"AB"	1	"BC"
"z"	1	"a"
"a B z"	4	"e F d"

---

answer

alpha = "abcdefghijklmnopqrstuvwxyz" #변수에 알파벳을 담아준다.
upalpha = alpha.upper() #대문자를 담을 변수를 만들고 alpha를 대문자로 변환해준다.
answer = "" #답이들어갈 변수
s = input("알파벳 입력\n") #알파벳을 입력받을 변수
n = input("밀 숫자 입력\n") #밀 숫자를 입력받을 변수
chk = True #대소문자 체크를 위한 변수
n = int(n) #밀 숫자를 정수형으로 변환
j = " " #공백을 구분 할 변수
for i in s: #입력받은 알파벳을 하나씩 반복문을 돌린다.
    if i == j:
        answer += i #i의 값이 공백일경우 answer에 공백을 더한다.
    elif i.islower() == chk: #i의 값이 소문자일경우
        pos = alpha.find(i) #alpha에서 i의 위치를 찾아 pos라는 변수에 담는다.
        if pos + n >= len(alpha): #pos의 숫자와 밀숫자를 더한값이 알파벳의 길이보다 클경우
            answer += alpha[(pos+n)%len(alpha)] #더한값에서 알파벳의 길이를 나눈 나머지를 answer에 더한다.
        else: #알파벳의 길이보다 작을경우
            answer += alpha[pos+n] #해당 인덱스의 알파벳을 answer에 더한다.
    else: #i의 값이 대문자일경우
        pos = upalpha.find(i) #upalpha에서 i의 위치를 찾아 pos에 담는다.
        if pos + n >= len(upalpha): #위와 동일
            answer += upalpha[(pos+n)%len(upalpha)]
        else:
            answer += upalpha[pos+n]
print (answer)

알고리즘 문제를 풀고 검색을 해보면 언제나 제 풀이보다 더욱 효율적인 방법이 있네요

언제쯤 다른 고수분들의 답과 비슷해질수 있을까요

열심히 공부해서 더욱 좋은 코드를 짜보도록 하겠습니다.

 

다풀고 나니까 함수로 만들질 않았네요