코딩테스트

시저 암호

개발천재나천재 2023. 1. 9. 10:16

 

문제 

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

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

 

 

 

풀이

"""
ABCDEFG .... Z
"""
from string import ascii_lowercase
from string import ascii_uppercase

def solution(s, n):
    answer = ''
    ans_list =[]
    
    alphabet_list = list(ascii_lowercase)
    alphabet_list_u = list(ascii_uppercase)
    
    s_list = list(map(str, s))

    for elem in s_list:
        if (elem.isupper()):
            idx = alphabet_list_u.index(elem)
            idx = (idx + n) % 26
            ans_list.append(alphabet_list_u[idx].upper())
        elif (elem.islower()):
            idx = alphabet_list.index(elem)
            idx = (idx + n) % 26
            ans_list.append(alphabet_list[idx].lower())
        else:
            ans_list.append(" ")
    
    answer = "".join(ans_list)
    
    return answer

 

 

출처 : https://school.programmers.co.kr/learn/courses/30/lessons/12926#