题目
实现一个 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元素的set方法没写,直接标注为public了。
前缀树的这个思想要记住!在很多应用场景都可以用得到。
具体场景可以看我之前的博客🔗:https://blog.csdn.net/qq_42670493/article/details/105164159
class Trie {
TrieNode root;
/** Initialize your data structure here. */
public Trie() {
root = new TrieNode();
}
/** Inserts a word into the trie. */
public void insert(String word) {
TrieNode cur = root;
for(int i = 0; i < word.length(); i++){
char c = word.charAt(i);
if(cur.children[c - 'a'] == null){
cur.children[c - 'a'] = new TrieNode(c);
}
cur = cur.children[c - 'a'];
//最后一个字母,加上结束标志
if(i == word.length() - 1){
cur.isLastWord = true;
}
}
}
/** Returns if the word is in the trie. */
public boolean search(String word) {
TrieNode cur = root;
for(int i = 0; i < word.length(); i++){
char c = word.charAt(i);
if(cur.children[c - 'a'] == null){
return false;
}
cur = cur.children[c - 'a'];
}
return cur.isLastWord;
}
/** Returns if there is any word in the trie that starts with the given prefix. */
public boolean startsWith(String prefix) {
TrieNode cur = root;
for(int i = 0; i < prefix.length(); i++){
char c = prefix.charAt(i);
if(cur.children[c - 'a'] == null){
return false;
}
cur = cur.children[c - 'a'];
}
return true;
}
}
class TrieNode{
char val;
public boolean isLastWord = false;
TrieNode[] children = new TrieNode[26];
public TrieNode(){}
public TrieNode(char val){
this.val = val;
}
}
/**
* Your Trie object will be instantiated and called as such:
* Trie obj = new Trie();
* obj.insert(word);
* boolean param_2 = obj.search(word);
* boolean param_3 = obj.startsWith(prefix);
*/
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/implement-trie-prefix-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。