LeetCode-208(medium)实现Trie前缀树

实现一个 Trie (前缀树),包含 insertsearch, 和 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 构成的。
  • 保证所有输入均为非空字符串。

这个构成的一个前缀树,就是每个word放到树里面,根节点是没有字符的!

然后,比如存一个apple,先将a放到子节点,然后p再放到下一个子节点,以此类推

在寻找的时候也是如此,顺着树从上往下找,就大约是下图这样的,如果再放'key'这个word,就在右边新建一个分支

LeetCode-208(medium)实现Trie前缀树_第1张图片

class TrieNode:
        # Initialize your data structure here.
        def __init__(self):
            self.word=False            #初始化什么都没有的
            self.children={}
    
class Trie:
        def __init__(self):
            self.root = TrieNode()
    
        # @param {string} word
        # @return {void}
        # Inserts a word into the trie.
        def insert(self, word):        #插入单词,是不用返回值的
            node=self.root
            for i in word:
 #每个字符插入的,这是为了找是否之前已经插入过同样的word,如果没有则需要开辟新的数,如果有就下移到子节点
                if i not in node.children:             
                     node.children[i]=TrieNode()  
                node=node.children[i]     # 当前节点到子节点!每次都要到这一步
            node.word=True
    
        # @param {string} word
        # @return {boolean}
        # Returns if the word is in the trie.
        def search(self, word):      #查找整个单词
            node=self.root
            for i in word:
                if i not in node.children:     # 每个字符对比,如果不在就false
                    return False
                node=node.children[i]     # 顺着树往下查找
            return node.word
    
        # @param {string} prefix
        # @return {boolean}
        # Returns if there is any word in the trie
        # that starts with the given prefix.
        def startsWith(self, prefix):     # 查找任意单词里的开头字符
            node=self.root
            for i in prefix:
                if i not in node.children:
                    return False
                node=node.children[i]
            return True
            
    
    # Your Trie object will be instantiated and called as such:
    # trie = Trie()
    # trie.insert("somestring")
    # trie.search("key")

 

你可能感兴趣的:(LeetCode)