常见面试算法题-快速人名查找

■ 题目描述

【快速人名查找】

给一个字符串，表示用’, \’分开的人名。然后给定一个字符串，进行快速人名查找，符合要求的输出。

快速人名查找要求︰人名的每个单词的连续前几位能组成给定字符串，一定要用到每个单词。

输入描述：

第一行是人名，用’, \’分开的人名
第二行是 查找字符串
输出描述：输出满足要求的人名

示例1   输入输出示例仅供调试，后台判题数据一般不包含示例

输入

zhang san,zhang san san
zs

输出

zhang san

示例2   输入输出示例仅供调试，后台判题数据一般不包含示例

输入

zhang san san,zhang an sa,zhang hang,zhang seng,zhang sen a
zhas

输出

zhang an sa,zhang seng

 以下代码为本人原创，可以供大家参考，若有不足之处，感谢指出！！！！

nums = input().split(",")
query = input()
rsp = []
for name in nums:
    words = name.split()
    if len(words) == 2:
        if words[0][0] != query[0]:
            continue
        list1 = [n for n, x in enumerate(query) if x == words[1][0]]
        for i in list1:
            if words[0].startswith(query[:i]) and words[1].startswith(query[i:]):
                rsp.append(name)
    if len(words) == 3:
        if words[0][0] != query[0]:
            continue
        list1 = [n for n, x in enumerate(query) if x == words[1][0]]
        list2 = [n for n, x in enumerate(query) if x == words[2][0]]
        for i in list1:
            for j in list2:
                if j > i:
                    if words[0].startswith(query[:i]) and words[1].startswith(query[i:j]) and words[2].startswith(query[j:]):
                        rsp.append(name)
print(','.join(rsp))