PTA jmu-python-凯撒密码加密算法

编写一个凯撒密码加密程序，接收用户输入的文本和密钥k，对明文中的字母a-z和字母A-Z替换为其后第k个字母。

输入格式:

接收两行输入，第一行为待加密的明文，第二行为密钥k。

输出格式:

输出加密后的密文。

输入样例:

在这里给出一组输入。例如：

Hello World!
3

输出样例:

在这里给出相应的输出。例如：

Khoor Zruog!

 代码实现：

a = input()
b = int(input())
c = ''
for i in a:
    if i.isalpha():
        temp = ord(i)+b
        if chr(temp).isalpha():
            c += chr(temp)
        else:
            c += chr(temp-26)
    else:
        c += i
print(c)