leetcode —— 面试题 17.多次搜索

给定一个较长字符串big和一个包含较短字符串的数组smalls，设计一个方法，根据smalls中的每一个较短字符串，对big进行搜索。输出smalls中的字符串在big里出现的所有位置positions，其中positions[i]为smalls[i]出现的所有位置。

示例：

输入：
big = “mississippi”
smalls = [“is”,“ppi”,“hi”,“sis”,“i”,“ssippi”]
输出： [[1,4],[8],[],[3],[1,4,7,10],[5]]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/multi-search-lcci
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
——————————
解题思路：将big的所有出现的字母的索引位置放在字典中，对于smalls中的每个单词，通过其首字母找到big中对应字母的起始位置，比较两个字符串是否相等，相等则进行记录。

其Python代码如下：

class Solution:
    def __init__(self):
        self.ans = []
    def multiSearch(self, big: str, smalls: List[str]) -> List[List[int]]:
        if not big:
            return [[] for _ in range(len(smalls))]
        dicts = dict()
        length = len(big)
        for i in range(length):  # 记录big中每个字母可能出现的索引
            dicts.setdefault(big[i],[])
            dicts[big[i]].append(i)
        
        for s in smalls: 
            lens = len(s)
            if lens == 0 or s[0] not in dicts.keys():  # 如果small中字符串的长度为零，或者字符串的首字母不在big中出现，直接返回空列表
                self.ans.append([])
                continue
            lists = []
            for index in dicts[s[0]]:  # 如果small的单词的首字母出现在big中，找到big中对应字母的所有出现位置
                if index+lens-1<length and big[index:index+lens]==s:  # 比较两个字符串是否相等
                    lists.append(index)
            self.ans.append(lists[:])  # 将记录存放在ans列表中
        return self.ans