用Java实现一个Trie树,trie,又称前缀树或字典树,是一种有序树,用于保存关联数组,其中的键通常是字符串。与二叉查找树不同,键不是直接保存在节点中,而是由节点在树中的位置决定。一个节点的所有子孙都有相同的前缀,也就是这个节点对应的字符串,而根节点对应空字符串。一般情况下,不是所有的节点都有对应的值,只有叶子节点和部分内部节点所对应的键才有相关的值。
更多关于Trie的介绍可以查看维基百科Trie树。
Trie的主要操作有,查找,插入等,下面是一个Trie树的实现
public class Trie {
TrieNode root; //这是一个根节点,根节点中没有字符
Trie(){ //构造函数
root = new TrieNode();
}
//查找trie树中是否存在某个单词
public boolean search(String word){
TrieNode temp = root;
for(char c:word.toCharArray()){
temp = temp.children[c-'a'];
if(temp==null){
return false;
}
}
if(temp.word.equals(word)){ //虽然前面单词都找到了,也可能不存在这个单词
return true;
}else{
return false;
}
}
public boolean startWith(String word){
TrieNode temp = root;
for(char c:word.toCharArray()){
if(temp.children[c-'a']==null){
return false;
}
temp = temp.children[c-'a'];
}
return true;
}
public void insert(String word){
TrieNode temp = root;
for(char c:word.toCharArray()){
if(temp.children[c-'a'] == null){
temp.children[c-'a'] = new TrieNode(c);
}
temp = temp.children[c-'a'];
}
temp.word = word;
}
class TrieNode {
TrieNode[] children; //每一个节点有26个孩子,每个孩子安装字典顺序进行排列,'a'排在第一个,'b'排在第二个,...,'z'排在最后一个
String word = ""; //到这个节点为止表示的单词,如果没有则为""
char c; //TrieNode中的字符
TrieNode(){
children = new TrieNode[26];
}
TrieNode(char c){
this.c = c;
children = new TrieNode[26];
}
}
}