Trie:
是一种多叉树,专门用来处理字符串。
查询每个条目的时间复杂度与条目数量无关,只与待查询的单词长度有关。
package Trie;
/*
工作过程:先定义一个遍历节点cur,将待查或待插入的单词字符取出,依次与cur.next比较
若不包含?新建一个节点并把这个字符存进去。cur.nexts.put(c,new Node());
若包含?就不用创建新节点,cur后移。cur=cur.nexts.get©;
*/
import java.util.TreeMap;
public class Trie{
class Node{//内部类节点
boolean isWord; //判断当前几个字母是否组成一个单词
TreeMap<Character,Node> nexts;//指向下个节点的指针
public Node(boolean isWord){//判断是否组成一个单词
this.isWord=isWord;
nexts=new TreeMap<Character,Node>();
}
public Node(){
this(false);
}
}
//成员变量
private Node root;//根节点
private int size;//元素个数
public Trie(){
root=new Node();
size=0;
}
//添加元素
public void add(String word){
Node cur=root;//创建一个遍历节点cur从root开始;
for(int i=0;i<word.length();i++){
char c=word.charAt(i);//将word中字符挨个取出,再查询遍历节点中是否包含;
if(cur.nexts.get(c)==null){
//如果遍历节点的下一个不包含c,就新建一个节点并把这个字符存进去
cur.nexts.put(c,new Node());
}
cur=cur.nexts.get(c);//如果包含,就不用创建新节点,cur后移
}
//避免重复
if(!cur.isWord){
cur.isWord=true;
size++;
}
}
//查询单词
public boolean contains(String word){
Node cur=root;
for(int i=0;i<word.length();i++){
char c=word.charAt(i);//依次取出word中的字符,再查询遍历节点中是否包含
if(cur.nexts.get(c)==null){
//如果遍历节点的下一个不包含该字符,返回false;
return false;
}
cur=cur.nexts.get(c);//如果包含,cur继续后移,查询下一个字符。
}
return cur.isWord;//最终返回是否组成单词
}
//查前缀
public boolean isPrefix(String prefix){
Node cur=root;
for(int i=0;i<prefix.length();i++){
char c=prefix.charAt(i);
if(cur.nexts.get(c)==null){
//如果遍历节点的下一个不包含c,返回false;
return false;
}
cur=cur.nexts.get(c);
}
return true;
}
}