前缀树简单使用

前缀树

每个结点由 int pass ; int end ; node [] nexts = new node[26]; 构成

Pass:表示此结点被通过的次数

End:表示路线在此节点结束

Nexts:表示此结点后面的结点，26对应26个字母，一开始为空，当有路线会通过结点时会新建出来. 比如说 有一个b字母在此结点的后面  则nexts[1] = new node(); nexts[1].pass++;

构建前缀树有什么用？

其中一个最重要的作用是：可以判断以路线为前缀的字符串有多少个。

比如说 ： 此时前缀树中初始化了  “ab”  “abc”  “abcd”

都通过了a，a的pass为3.那么以a为前缀的字符串有3个

---

package xyz.yeidi.tree;

//前缀树
/*
* 1.实现查找字符串的功能 serach
* 2.实现得到以某字符串为前缀的字符串个数的功能 getNumByPrefix
*/
public class PrefixTree {
//前缀树的根结点
private static Node root = new Node();
public static void main(String[] args) {
String [] arrs = new String [] {"abc","ab","bc","ionxks","ab","abc","abc"};
initTree(arrs);
System.out.println(search("c"));
System.out.println(getNumByPrefix("ab"));
}

//初始化前缀树
public static void initTree(String []arrs) {
Node node = root;
int index = 0; //用于表示下标与字母的对应关系
for(int i=0;i<arrs.length;i++) {
char[] array = arrs[i].toCharArray();
root.pass++;
node = root;
for(int j=0;j<array.length;j++) {
index = array[j] - 'a';
if(node.nexts[index]==null) {
node.nexts[index] = new Node();
}
node.nexts[index].pass++;
node = node.nexts[index];
}
node.end++;
}
}

//查找是否含有字符串str,且返回其含有几个
public static int search(String str) {
//如果传入的字符串为null则返回0
if(str == null) {
return 0;
}
Node node = root;
char[] arr = str.toCharArray();
int index = 0;
for(int i=0;i<arr.length;i++) {
index = arr[i] - 'a';
if(node.nexts[index]==null) {
return 0;
}
node = node.nexts[index];
}
return node.end;
}

//得到以某字符串为前缀的字符串个数
public static int getNumByPrefix(String str) {
//如果传入的字符串为null则返回0
if(str == null) {
return 0;
}
Node node = root;
char[] arr = str.toCharArray();
int index = 0;
for(int i=0;i<arr.length;i++) {
index = arr[i] - 'a';
if(node.nexts[index]==null) {
return 0;
}
node = node.nexts[index];
}
return node.pass;
}
}
//1.0：只有小写字母的版本
class Node {
int pass = 0;
int end = 0;
Node [] nexts; //2.0版本 将Node[]nexts 改成hash表 HashMap()这样我们就不单单应用于小写字母
public Node() {
nexts = new Node [26]; //对应26个字母
}
}