【华为机试题】 [Python] 英文输入法

题目

"""
问题描述
K小姐是一家科技公司的产品经理,她最近负责开发一款英文输入法。其中一个重要功能是单词联想,即根据用户输入的单词前缀,从已输入的英文语句中联想出用户想输入的单词,并按字典序输出联想到的单词序列。如果无法联想到任何单词,则输出用户输入的单词前缀。

在实现这个功能时,需要注意以下几点:

英文单词联想时区分大小写。
缩略形式如 "don't" 应被视为两个单词 "don" 和 "t"。
输出的单词序列中不能有重复单词,且只能包含英文单词,不能包含标点符号。
输入格式
输入包含两行:

第一行是一个由英文单词word
word 和标点符号组成的语句str

第二行是一个英文单词前缀𝑝𝑟𝑒

输出格式
输出符合要求的单词序列或单词前缀,如果有多个单词,则用单个空格分隔。

样例输入1
I love you
He
样例输出1
He

样例输入2
The furthest distance in the world, ls not between life and death, But when I stand in front of you, Yet you don't know that I love you.
f
样例输出2
front furthest
"""

代码

class Solution:
    def func(self, word, pre):
        ans_list = []
        words = word.replace('\'', ' ').split()
        for single_word in words:
            if single_word.startswith(pre):
                ans_list.append(single_word)
        return pre if not ans_list else ans_list

s = Solution()
# word = "I love you"
# pre = "He"
word = "The furthest distance in the world, ls not between life and death, But when I stand in front of you, Yet you don't know that I love you."
pre = "f"
ans = s.func(word, pre)
print(ans)