题解 | 简单密码

import sys, re

for line in sys.stdin:
    a = line.strip().split()[0]
    s_lst = []
    mapping = {'abc': 2, 'def':3, 'ghi':4, 'jkl':5, 'mno':6, 'pqrs':7, 'tuv':8, 'wxyz':9}

    for s in a:

        if s.islower():
            for key, num in mapping.items():
                if s in key:
                    s_lst.append(str(mapping[key]))

        if s.isupper():
            s = chr((ord(s.lower()) - ord('a') + 1) % 26 + ord('a')).lower()
            s_lst.append(s)
        

        if s.isdigit():
            s_lst.append(s)
    print(''.join(s_lst))