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

코딩테스트연습

• programmers.co.kr
• 시저암호

 

---

 

def solution(s, n):
    #lowers = 97~122
    #uppers = 65~90
    answer = ""

    for x in s:
        if 65 <= ord(x) <=90:
            if ((ord(x)+n) > 90):
                answer += chr(65+((ord(x)+n)-91))
            else:
                answer += chr(ord(x)+n)
        elif 97 <= ord(x) <=122:
            if ((ord(x)+n) >122):
                answer += chr(97+((ord(x)+n)-123))
            else:
                answer += chr(ord(x)+n)
        else:
            answer += x
    return answer