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 次调用

解题思路

哈希暴力，其实也可以字典树，但是思路不太想得明白

代码

Python

class WordFilter:

    def __init__(self, words: list[str]):
        self.map = {}
        for i, word in enumerate(words):
            for j in range(1, len(word) + 1):
                for k in range(1, len(word) + 1):
                    self.map[(word[:j], word[-k:])] = i

    def f(self, pref: str, suff: str) -> int:
        return self.map.get((pref, suff), -1)