题解 | #简单密码#

import sys


def get_num(key):
    if key in 'abc':
        return 2
    elif key in 'def':
        return 3
    elif key in 'ghi':
        return 4
    elif key in 'jkl':
        return 5
    elif key in 'mno':
        return 6
    elif key in 'pqrs':
        return 7
    elif key in 'tuv':
        return 8
    elif key in 'wxyz':
        return 9

    
def get_letter(key):
    letters = 'abcdefghijklmnopqrstuvwxyz'
    for index, letter in enumerate(letters):
        if letter == key:
            if index == len(letters) - 1:
                return 'a'
            else:
                return letters[index + 1]


for line in sys.stdin:
    for content in line:
        if content.islower():
            print(get_num(content), end='')
        elif content.isupper():
            print(get_letter(content.lower()), end='')
        else:
            print(content, end='')