7-30 jmu-python-凱撒密碼加密算法 (10 分)

編寫一個凱撒密碼加密程序，接收用戶輸入的文本和密鑰k，對明文中的字母a-z和字母A-Z替換為其后第k個字母。

輸入格式:

接收兩行輸入，第一行為待加密的明文，第二行為密鑰k。

輸出格式:

輸出加密后的密文。

輸入樣例:

在這里給出一組輸入。例如：

Hello World!
3

輸出樣例:

在這里給出相應的輸出。例如：

Khoor Zruog!

s1 = ['a','b','c','d','e','f','g','h','i','j','k','l','m','n','o','p','q','r','s','t','u','v','w','x','y','z']
s2 = ['A','B','C','D','E','F','G','H','I','J','K','L','M','N','O','P','Q','R','S','T','U','V','W','X','Y','Z']
s = input()
k = int(input())
for i in range(len(s)):
    if 'a' <= s[i] <= 'z':
        for j in range(26):
            if s1[j] == s[i]:
                print("{}".format(s1[(j + k) % 26]), end="")
    elif 'A' <= s[i] <= 'Z':
        for j in range(26):
            if s2[j] == s[i]:
                print("{}".format(s2[(j + k) % 26]), end="")
    else:
        print("{}".format(s[i]),end="")