性质:
时间复杂度为O(w);w为查询单词的长度
import java.util.TreeMap;
public class Trie {
private class Node{
public boolean isWord;
public TreeMap<Character, Node> next;
public Node(){
this(false);
}
public Node( boolean isWord){
this.isWord = isWord;
next = new TreeMap<>();
}
}
private Node root;
private int size;
public Trie(){
root = new Node();
size = 0;
}
public int getSize(){
return size;
}
//向Trie中添加新单词word
public void add(String word){
Node cur = root;
for (int i = 0; i < word.length(); i++) {
char c = word.charAt(i);
if(null == cur.next.get(c)){
cur.next.put(c, new Node());
}
cur = cur.next.get(c);
}
if(!cur.isWord){
cur.isWord = true;
size++;
}
}
//查询Trie中是否有单词word
public boolean contains(String word){
Node cur = root;
for (int i = 0; i < word.length(); i++) {
char c = word.charAt(i);
if (null == cur.next.get(c)) {
return false;
}
cur = cur.next.get(c);
}
return cur.isWord;
}
//查询Trie中是否以prefix为前缀
public boolean isPrefix(String prefix){
Node cur = root;
for (int i = 0; i < prefix.length(); i++) {
char c = prefix.charAt(i);
if (null == cur.next.get(c)) {
return false;
}
cur = cur.next.get(c);
}
return true;
}
}