题解 | 字符串加解密

# Process input
s, t = input(), input()

# Algo
ord_a = ord('a')
ord_A = ord('A')
ord_0 = ord('0')
def encode(string, offsetFunc):
    for ch in string:
        if ch.isdigit():
            base = ord_0
            newBase = ord_0
            cnt = 10
        elif ch.islower():
            base = ord_a
            newBase = ord_A
            cnt = 26
        elif ch.isupper():
            base = ord_A
            newBase = ord_a
            cnt = 26
        else:
            raise NotImplemented()
        chEnc = chr(newBase + (offsetFunc(ord(ch), 1) - base) % cnt)
        print(chEnc, end='')
    print()

# Encrypt
encode(s, lambda x, y: x + y)
# Decrypt
encode(t, lambda x, y: x - y)