请你设计一个数据结构,支持 添加新单词 和 查找字符串是否与任何先前添加的字符串匹配 。
实现词典类 WordDictionary :
WordDictionary() 初始化词典对象
void addWord(word) 将 word 添加到数据结构中,之后可以对它进行匹配
bool search(word) 如果数据结构中存在字符串与 word 匹配,则返回 true ;否则,返回 false 。word 中可能包含一些 ‘.’ ,每个 . 都可以表示任何一个字母。
示例:
输入:
[“WordDictionary”,“addWord”,“addWord”,“addWord”,“search”,“search”,“search”,“search”]
[[],[“bad”],[“dad”],[“mad”],[“pad”],[“bad”],[“.ad”],[“b…”]]
输出:
[null,null,null,null,false,true,true,true]
解释:
WordDictionary wordDictionary = new WordDictionary();
wordDictionary.addWord(“bad”);
wordDictionary.addWord(“dad”);
wordDictionary.addWord(“mad”);
wordDictionary.search(“pad”); // 返回 False
wordDictionary.search(“bad”); // 返回 True
wordDictionary.search(“.ad”); // 返回 True
wordDictionary.search(“b…”); // 返回 True
提示:
1 <= word.length <= 25
addWord 中的 word 由小写英文字母组成
search 中的 word 由 ‘.’ 或小写英文字母组成
最多调用 104 次 addWord 和 search
支持.
的前缀树
class WordDictionary:
def __init__(self):
self.is_end = False
self.children = [None] * 26
def __mapping(self, ch):
return ord(ch) - ord("a")
def addWord(self, word: str) -> None:
node = self
for c in word:
if not node.children[self.__mapping(c)]:
node.children[self.__mapping(c)] = WordDictionary()
node = node.children[self.__mapping(c)]
node.is_end = True
def search(self, word: str) -> bool:
node = self
c, left = word[:1], word[1:]
if c != ".":
child = node.children[self.__mapping(c)]
if not child:
return False
elif not left:
return child.is_end
else:
return child.search(left)
else:
res = False
for idx, child in enumerate(node.children):
if not child:
continue
res |= node.search(chr(idx + ord("a")) + left)
if res:
return res
return res