208 实现 Trie (字典树)
题目:
实现一个 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 构成的。
保证所有输入均为非空字符串。
其实这道题不难,最难的是要理解字典树的结构,如下图:
如果能理解字典树,就能比较简单的写出这道题,注意一点就是,如何到自己的孩子节点
初始化:就是对root初始化,并附上空值
插入:一个循环,判断这个节点有没有值,没有就插入,然后将指针移动到自己的孩子节点,继续
查找:同样的循环,一个一个字符查找,如果一旦有为空的,就直接返回false,如果遍历完了没错,再判断isWord是否为真,看是不是已经到了结尾
查询部分:就跟之前不同的是不用最后判断是否是整字符
代码:
public class TrieNode{
public char val;
public boolean isWord;
public TrieNode[] children = new TrieNode[26];
public TrieNode(){}
TrieNode(char c){
TrieNode node = new TrieNode();
node.val = c;
}
}
private TrieNode root;
public Trie() {
root = new TrieNode();
root.val=' ';
}
/** Inserts a word into the trie. */
public void insert(String word) {
TrieNode ws = root;
for (int i = 0; i <word.length() ; i++) {
char c = word.charAt(i);
if(ws.children[c-'a']==null){
ws.children[c-'a']= new TrieNode(c);
}
ws=ws.children[c-'a'];//往下走
}
ws.isWord=true;
}
/** Returns if the word is in the trie. */
public boolean search(String word) {
TrieNode ws = root;
for (int i = 0; i <word.length() ; i++) {
char c = word.charAt(i);
if(ws.children[c-'a']==null) return false;
ws=ws.children[c-'a'];//往下走
}
return ws.isWord;//判断是否是单词的结尾,而不是一部分
}
/** Returns if there is any word in the trie that starts with the given prefix. */
public boolean startsWith(String prefix) {
TrieNode ws = root;
for (int i = 0; i <prefix.length() ; i++) {
char c = prefix.charAt(i);
if(ws.children[c-'a']==null) return false;
ws=ws.children[c-'a'];//往下走
}
return true;
}