[7.4] 시저 암호 알고리즘 (49/100)

코딩테스트 연습 - 시저 암호 | 프로그래머스 (programmers.co.kr)

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

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

def solution(s, n):
    answer = ''
    upper = "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
    under = "abcdefghijklmnopqrstuvwxyz"
    for i in range(len(s)):
      if s[i].isupper():
        save = upper.find(s[i]) + n
        answer += upper[(save)%len(upper)]
      elif s[i].islower():
        save = under.find(s[i]) + n
        answer += under[(save)%len(under)]
      else:
        answer += ' '
    return answer