题解 | 字符串加解密

import sys

su = [str(_) for _ in range(9)]
nisu = [str(_) for _ in range(1,10)]
index = 0


def encrpty(s):
    res = ""
    for _ in s:
        if _ in su:
            temp = int(_) + 1
            res += str(temp)
        elif _ == "9":
            res += "0"
        elif _ == "Z":
            res += "a"
        elif _ == "z":
            res += "A"
        else:
            # 字符
            if _.lower() != _:
                flag = True
            else:
                flag = False
            temp = chr(ord(_) + 1)
            if flag:
                res += temp.lower()
            else:
                res += temp.upper()
    return res


def decrpty(s):
    res = ""
    for _ in s:
        if _ in nisu:
            temp = int(_) - 1
            res += str(temp)
        elif _ == "0":
            res += "9"
        elif _ == "a":
            res += "Z"
        elif _ == "A":
            res += "z"
        else:
            # 字符
            if _.lower() != _:
                flag = True
            else:
                flag = False
            temp = chr(ord(_) - 1)
            if flag:
                res += temp.lower()
            else:
                res += temp.upper()
    return res


for line in sys.stdin:
    a = line.split()[0]
    if index == 0:
        print(encrpty(a))
    else:
        print(decrpty(a))
    index += 1

模拟