프로그래머스 - 시저암호

서유진·2022년 3월 24일
0

미숙한 제 풀이는 정답이 될 수 없으며, 이것보다 효율적인 코드는 당연히 존재합니다.
참고만 하여 주시고, 관련 피드백은 항상 환영합니다. 🤍

import string

def solution(s, n):
    answer = []
    lower = [x for x in string.ascii_lowercase]
    upper = [x for x in string.ascii_uppercase]

    for a in range(len(s)):
        if s[a] == ' ':
            answer.append(' ')
            pass
        else:
            if s[a].isupper(): # 대문자일때
                if upper.index(s[a])+n >= len(upper):
                    temp_index = upper.index(s[a])+n - len(upper)
                else:
                    temp_index = upper.index(s[a])+n
                answer.append(upper[temp_index])
            else : #소문자일때
                if lower.index(s[a])+n >= len(lower):
                    temp_index = lower.index(s[a])+n - len(lower)
                else:
                    temp_index = lower.index(s[a])+n
                answer.append(lower[temp_index])

    return ''.join(answer)
    
profile
Backend Dev.

0개의 댓글