前缀树是啥,我们今天来聊聊。
一、前缀树定义
1)单个字符串中,字符从前到后的加到一棵多叉树上
2)字符放在边上,节点上有专属的数据项(常见的是pass和end值)
3)样本添加方式,每个字符串都从根节点开始加,如果没有路就新建,如果有路就复用
4)添加时,沿途节点的pass值加1,每个字符串结束时来到的节点end值加1
作用:可以更加方便的完成前缀相关的查询
二、简版的前缀树
只能存放26个小写字母组成的字符串
/**
* 字符种类:只有26个小写字母
*/
public static class Node1 {
// 当前节点被经过了几次
private int pass;
// 有多少个字符串以当前节点结尾
private int end;
// 当前节点所有的子节点
private Node1[] nexts;
// 节点只存放26个小写字母
public Node1() {
pass = 0;
end = 0;
// node[i] == null,节点不存在
nexts = new Node1[26];
}
}
public static class Trie1 {
private Node1 root;
public Trie1() {
root = new Node1();
}
/**
* 向前缀树中添加一个字符串
*/
public void insert(String word) {
if (word == null) {
return;
}
// 正在添加字符的节点
Node1 node = root;
// 已经有字符经过该节点,修改节点pass值
node.pass++;
int path = 0;
// 当前单词挨个字符的添加到前缀树上
char[] chars = word.toCharArray();
for (char aChar : chars) {
// 当前字符应该走的路径
path = aChar - 'a';
// 当前节点的path路径对应节点不存在,则新建
if (node.nexts[path] == null) {
node.nexts[path] = new Node1();
}
// 当前节点的path路径对应节点肯定已经存在了
// 所以当前节点来到path路径对应节点
node = node.nexts[path];
// 已经有字符到达该节点,修改pass值
node.pass++;
}
// 字符添加完毕,修改最后节点的end值
node.end++;
}
/**
* word单词之前加过多少次
*/