题解 | 简单密码

def string_shift(s):
    res = []
    l = len(s)
    base = ord('A')
    for i in range(l):
        if not s[i].isalpha():
            res.append(s[i])
        else:
            shifted = chr( (ord(s[i]) - base -5) % 26 + base)
            res.append(shifted)
    res = ''.join(res)
    return res


while True:
    try:
        s = input()
        if s == 'START' or s == 'END':
            continue
        elif s == 'ENDOFINPUT':
            break
        else:
            s_list = s.split(' ')
            res = []
            for i in s_list:
                #print(string_shift(i))
                res.append(string_shift(i))
        res = ' '.join(res)
        # print(res)
        print(res)
    except:
        break