[LeetCode解题报告] 211. 添加与搜索单词 - 数据结构设计

一、 题目

1. 题目描述

请你设计一个数据结构，支持 添加新单词 和 查找字符串是否与任何先前添加的字符串匹配 。

实现词典类 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

Related Topics

• 深度优先搜索
• 设计
• 字典树
• 字符串

• 426
• 0

2. 原题链接

链接: 211. 添加与搜索单词 - 数据结构设计

二、 解题报告

1. 思路分析

非常常规的字典树

2. 复杂度分析

最坏时间复杂度O(nlog₂n)

3.代码实现

字典树用dfs匹配.。

class TrieNode:
    def __init__(self,cnt=0):
        self.cnt = cnt 
        self.next = [None]*26
        self.is_end = False
    
class WordDictionary:
    def __init__(self):
        self.root = TrieNode()
        self.depth = 0

    def addWord(self, word: str) -> None:
        cur = self.root
        for c in word:
            i = ord(c)-ord('a')
            if not cur.next[i] :  # 没有这个字符
                cur.next[i] = TrieNode()
            cur = cur.next[i]
            cur.cnt += 1
        cur.is_end = True            

    def search(self, word: str) -> bool:
        def dfs(root,word,start):
            if not root:
                return False
            if start == len(word):
                return root.is_end

            c = word[start]
            if c == '.':
                for i in range(26):
                    if root.next[i] and root.next[i].cnt >0 and dfs(root.next[i],word,start+1):
                        return True
                return False

            else :
                i = ord(c)-ord('a')
                if not root.next[i] or root.next[i].cnt == 0:
                    return False
                return dfs(root.next[i],word,start+1)
            
        return dfs(self.root,word,0)

三、 本题小结

1. 字典树可以用数组或树形结构实现，树形好写一些