【LeetCode】745. 前缀和后缀搜索

题目

745. 前缀和后缀搜索

设计一个包含一些单词的特殊词典，并能够通过前缀和后缀来检索单词。

实现 WordFilter 类：

• WordFilter(string[] words) 使用词典中的单词 words 初始化对象。
• f(string pref, string suff) 返回词典中具有前缀 prefix 和后缀 suff 的单词的下标。如果存在不止一个满足要求的下标，返回其中 最大的下标 。如果不存在这样的单词，返回 -1 。

示例：

输入
["WordFilter", "f"]
[[["apple"]], ["a", "e"]]
输出
[null, 0]
解释
WordFilter wordFilter = new WordFilter(["apple"]);
wordFilter.f("a", "e"); // 返回 0 ，因为下标为 0 的单词：前缀 prefix = "a" 且 后缀 suff = "e" 。

提示：

• 1 <= words.length <= 104
• 1 <= words[i].length <= 7
• 1 <= pref.length, suff.length <= 7
• words[i]、pref 和 suff 仅由小写英文字母组成
• 最多对函数 f 执行 104 次调用

思路

• 将所有的前后缀组合保存在一个字典当中，访问则直接访问字典

代码

class WordFilter:

    def __init__(self, words: List[str]):
        self.dic = {}
        for i,word in enumerate(words):
            for pre in range(1,len(word)+1):
                for suff in range(1,len(word)+1):
                    self.dic[word[:pre] + "#" + word[-suff:]] = i
        

    def f(self, pref: str, suff: str) -> int:
        return self.dic.get(pref+"#"+suff,-1)

复杂度

• init
  • 时间复杂度：$O(lengt{h}_{words}\ast lengt{h}_{word}^2)$
  • 空间复杂度：$O(lengt{h}_{words}\ast lengt{h}_{word}^2)$
• f
  • 时间复杂度：$O(1)$
  • 空间复杂度：$O(1)$