暑假leetcode每日一题打卡-第七天-前缀和后缀搜索（hard）_wordfilter表示单词过滤器一个string类-CSDN博客

本文链接：https://blog.csdn.net/qq_51898139/article/details/125784354

前言：

通过leetcode记录自己每天坚持刷题，以此监督自己的学习。不能放假了就懈怠😁

题目简介

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

实现 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” 。

解析

刚开始误解了题目的意思，以为prefix和suff都只有一个字符，所以就写了个错误的版本，

class WordFilter:

    def __init__(self, words: List[str]):
        self.words=words
        self.dic={}
        for i in range(len(words)):
            pre,suff=self.words[i][0]+self.words[i][-1]
            self.dic[(pre,suff)]=i # 这里用元组做key，若用string：“ab"+”c"和"a"+"bc"就无法分辨了


    def f(self, pref: str, suff: str) -> int:
        if (pref,suff) in self.dic:
            return self.dic[(pref,suff)]
        else:
            return -1

我的想法是把words中前后缀记录下来放在map中，有重复的还可以直接覆盖,然后直接看前后缀在字典中的位置就可以了

但是没想到pref和suff没有规定字符个数
但是我感觉也差不多是这个思想，然后我就枚举所有前后缀的可能放在map中，结果提交成功了

代码

class WordFilter:

    def __init__(self, words: List[str]):
        wordsDic={}
        for i, w in enumerate(words):
            for j in range(0, len(w) + 1):
                p = w[:j]
                for k in range(0, len(w) + 1):
                    q = w[k:]
                    wordsDic[(p,q)] = i
        self.wordsDic=wordsDic

    def f(self, pref: str, suff: str) -> int:
        if (pref,suff) in self.wordsDic:
            return self.wordsDic[(pref,suff)]
        else:
            return -1