题解 | 简单密码

import sys

for line in sys.stdin:
    a = line.split()
    # print(int(a[0]) + int(a[1]))
    password = a[0]
    trans = []
    # trans_dict = { #太麻烦
    #     2:['a','b','c'],
    #     3:['d','e','f'],
    #     4:['g','h','i'],
    #     5:['j','k','l'],
    #     6:['m','n','o'],
    #     7:['p','q','r','s'],
    #     8:['t','u','v'],
    #     9:['w','x','y','z']
    # }
    trans_dict = {
        "abc":2,
        "def":3,
        "ghi":4,
        "jkl":5,
        "mno":6,
        "pqrs":7,
        "tuv":8,
        "wxyz":9
    }
    # c_list = ['a','b','c','d','e','f','g','h','i','j','k','l','m','n','o','p','q','r','s','t','u','v','w','x','y','z']
    for c in password:
        if c.isdigit():
            trans.append(c)
        elif c.islower():
            for k,v in trans_dict.items():
                if c in k:
                    c_trans = str(v)
                    trans.append(c_trans)
        elif c.isupper():
            # idx = c_list.index(c.lower()) 
            # c_trans = c_list[(idx+1) % len(c_list)]
            # trans.append(c_trans)
            if c == 'Z':
                trans.append('a')
            else:
                trans.append(chr(ord(c.lower())+1))
    print(''.join(trans))