华为OD机试真题-英文输入法--Python-OD统一考试（C卷）

题目描述：

主管期望你来实现英文输入法单词联想功能。需求如下：

依据用户输入的单词前缀，从已输入的英文语句中联想出用户想输入的单词，按字典序输出联想到的单词序列，如果联想不到，请输出用户输入的单词前缀。

注意：

1.  英文单词联想时，区分大小写

2.  缩略形式如”don't”，判定为两个单词，”don”和”t”

3.  输出的单词序列，不能有重复单词，且只能是英文单词，不能有标点符号

输入描述：

输入为两行。

首行输入一段由英文单词word和标点符号组成的语句str；

接下来一行为一个英文单词前缀pre。

0 < word.length() <= 20

0 < str.length <= 10000

0 < pre <= 20

输出描述：

输出符合要求的单词序列或单词前缀，存在多个时，单词之间以单个空格分割

示例1

输入：

I love you
He
输出：

He
说明：

从用户已输入英文语句”I love you”中提炼出“I”、“love”、“you”三个单词，接下来用户输入“He”，从已输入信息中无法联想到任何符合要求的单词，因此输出用户输入的单词前缀。

示例2

输入：

The furthest distance in the world, Is not between life and death, But when I stand in front of you, Yet you don't know that I love you.
f
输出：

front furthest
说明：

从用户已输入英文语句”The furthestdistance in the world, Is not between life and death, But when I stand in frontof you, Yet you dont know that I love you.”中提炼出的单词，符合“f”作为前缀的，有“furthest”和“front”，按字典序排序并在单词间添加空格后输出，结果为“frontfurthest”。
 

解题思路：考察对数组、字符串、 排序、 循环的理解。

def remove(str):
    return str.replace(".", "")

def compute(str, pre):
    set = set()
    arr = str.split(", ")
    for i in range(len(arr)):
        temp = arr[i]
        arr2 = temp.split(" ")
        if len(arr2) == 1:
            set.add(remove(arr2[0]))
        else:
            for j in range(len(arr2)):
                temp2 = arr2[j]
                arr3 = temp2.split("'")
                if len(arr3) == 1:
                    set.add(remove(arr3[0]))
                else:
                    for s in arr3:
                        set.add(remove(s))

    res = set()
    for s in set:
        if s.startswith(pre):
            res.add(s)

    if len(res) == 0:
        print(pre)
    else:
        sb = ""
        for s in res:
            sb += " " + s

        print(sb[1:])

def main():
    n = 2
    str = None
    pre = None
    while True:
        if n == 2:
            str = input()
        if n == 1:
            pre = input()

        n -= 1
        if n == 0:
            break

    compute(str, pre)

if __name__ == "__main__":
    main()