题解 | 字符串加解密

import re
import sys
dis = ord('a')-ord('A')
def codecha(c):
    if c.isalpha():
        if c.islower():
            res = chr(ord(c)+1-dis) if c!='z' else 'A'
        elif c.isupper():
            res = chr(ord(c)+1+dis) if c!='Z' else 'a'
    elif c.isdigit():
        res = chr(ord(c)+1) if c!='9' else '0'
    else:
        res = c
    return res


def decodecha(c):
    if c.isalpha():
        if c.islower():
            res = chr(ord(c)-1-dis) if c!='a' else 'Z'
        elif c.isupper():
            res = chr(ord(c)-1+dis) if c!='A' else 'z'
    elif c.isdigit():
        res = chr(ord(c)-1) if c!='0' else '9'
    else:
        res = c
    return res


s1 = input() 
s2 = input() 
res = []
for c in s1:
    res.append(codecha(c))
print(''.join(res))

res = []
for c in s2:
    res.append(decodecha(c))
print(''.join(res))