实现一个 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 构成的。
- 保证所有输入均为非空字符串。
这个构成的一个前缀树,就是每个word放到树里面,根节点是没有字符的!
然后,比如存一个apple,先将a放到子节点,然后p再放到下一个子节点,以此类推
在寻找的时候也是如此,顺着树从上往下找,就大约是下图这样的,如果再放'key'这个word,就在右边新建一个分支
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")