字典树

208. 实现 Trie (前缀树)

难度中等220收藏分享切换为英文关注反馈

实现一个 Trie (前缀树)，包含 insert, search, 和 startsWith 这三个操作。

示例:

Trie trie = new Trie();

trie.insert("apple");
trie.search("apple");   // 返回 true
trie.search("app");     // 返回 false
trie.startsWith("app"); // 返回 true
trie.insert("app");   
trie.search("app");     // 返回 true

说明:

• 你可以假设所有的输入都是由小写字母 a-z 构成的。
• 保证所有输入均为非空字符串。

class Trie:

    def __init__(self):
        """
        Initialize your data structure here.
        """
        self.root = {}


    def insert(self, word: str) -> None:
        """
        Inserts a word into the trie.
        """
        node = self.root
        for char in word:
            #setdefault()方法使用指定的键返回项目的值.如果键不存在，则插入这个具有指定值的键.
            node =  node.setdefault(char, {}) 

        node["end"] = True


    def search(self, word: str) -> bool:
        """
        Returns if the word is in the trie.
        """
        node = self.root
        for char in word:
            if char not in node:
                return False
            node = node[char]

        return "end" in node


    def startsWith(self, prefix: str) -> bool:
        """
        Returns if there is any word in the trie that starts with the given prefix.
        """
        node =  self.root
        for char in prefix:
            if char not in node:
                return False
            node = node[char]
        
        return True

插入apple与apply：

 {'a': {'p': {'p': {'l': {'e': {'end': True}}}}}}

 {'a': {'p': {'p': {'l': {'e': {'end': True}, 'y': {'end': True}}}}}}

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

难度中等98

设计一个支持以下两种操作的数据结构：

void addWord(word)
bool search(word)

search(word) 可以搜索文字或正则表达式字符串，字符串只包含字母 . 或 a-z 。 . 可以表示任何一个字母。

示例:

addWord("bad")
addWord("dad")
addWord("mad")
search("pad") -> false
search("bad") -> true
search(".ad") -> true
search("b..") -> true

说明:

你可以假设所有单词都是由小写字母 a-z 组成的。

class WordDictionary:

    def __init__(self):
        """
        Initialize your data structure here.
        """
        self.root = {}


    def addWord(self, word: str) -> None:
        """
        Adds a word into the data structure.
        """
        node = self.root
        for char in word:
            node = node.setdefault(char, {})
        
        node['end'] = True


    def search(self, word: str) -> bool:
        """
        Returns if the word is in the data structure. A word could contain the dot character '.' to represent any one letter.
        """
        return self._dfs(self.root, word)

    def _dfs(self, dictionary, word):

        for index, char in enumerate(word):
            if char == '.':
                res = 0
                for any_char in dictionary:
                    if any_char != 'end':
                        temp_res = self._dfs(dictionary[any_char], word[index+1:]) 
                        res += temp_res
                return res != 0

            elif char not in dictionary:
                return False
                
            else:
                dictionary = dictionary[char]

        return 'end' in dictionary