华为OD机试 Python 【查字典】

题目

描述：
给你一个“头”（就是词的开始部分）和一个单词清单。你需要找出哪些单词是以这个“头”开始的。

输出：

所有以这个“头”开始的单词，每个单词一行。
如果一个都没有，输出“-1”。

输入 abc 4 a ab abc abcd
输出
abc

abcd

思路

find_words_with_prefix 这个函数就是帮我们找头的，传入一个“头”和一个词列表，它就会帮你找出所有的匹配的词。
startswith 这个函数就是Python里判断一个字符串是不是以某个“头”开始的。通过split函数分割，然后拿到我们的“头”和词列表。最后，如果找到了匹配的词，就一个个打印出来；如果一个都没找到，那就打印-1。

代码

def find_words_with_prefix(prefix, word_list):
    result = []