Trie Tree 字典树 _字典树

（Trie Tree）字典树
场景：在n个字符串中查找某个字符串。
暴力匹配，时间复杂度为O（nm），m为字符串平均长度，效率过低。
字典查找单词"fly"，首先查找’f’,然后查找’l’,最后查找’y’,实现查找，字典树就是完成模拟查找过程。
例：
通过这6个单词构造字典树：
在每次单词的节点处设置标记，是否为单词结尾。

文章插图
字典树应用：
字符串检索统计单词出现的次数前缀匹配字符串排序
问题链接
问题：
最近遇到一个难题,老师交给他很多单词(只有小写字母组成,不会有重复的单词出现),现在老师要他统计出以某个字符串为前缀的单词数量(单词本身也是自己的前缀).
输入：
输入数据的第一部分是一张单词表,每行一个单词,单词的长度不超过10,它们代表的是老师交给统计的单词,一个空行代表单词表的结束.第二部分是一连串的提问,每行一个提问,每个提问都是一个字符串.
注意:本题只有一组测试数据,处理到文件结束.
输出：
对于每个提问,给出以该字符串为前缀的单词的数量.
输入样例：
bananabandbeeabsoluteacmbabbandabc
输出样例：
2310
使用Map：

import java.util.*;class Main {public static void main(String[] args) {Scanner in = new Scanner(System.in);//单词前缀MapMap prefix = new HashMap<>();while (true) {String s = in.nextLine();if (s.equals("")) {break;}//空行单词表结束for (int i = s.length(); i > 0; i--) {//向map里添加数据String tmp = s.substring(0, i);prefix.put(tmp, prefix.getOrDefault(tmp, 0) + 1);}}while (in.hasNext()) {String query = in.nextLine();System.out.println(prefix.getOrDefault(query, 0));}}}

使用数组构造字典树结构体：
【Trie Tree字典树】

import java.util.*;class Main {static int[][]trie=new int[1000010][26];//数组定义字典树，存储下一个字符的位置static int[]num=new int[1000010];//以某个字符为前缀的单词数量static int pos=1;//当前新分配存储数量/*** 向字典树中插入单词*/public static void insert(char[]str){int p=0;for (int i = 0; i < str.length; i++) {int n=str[i]-'a';if(trie[p][n]==0){trie[p][n]=pos++;}p=trie[p][n];num[p]++;}}/**** @paramstr 字符数组* @return以字符串为前缀的单词数量*/public static int find(char[] str){int p=0;for (int i = 0; i < str.length; i++) {int n=str[i] - 'a';if(trie[p][n] == 0)return 0;p=trie[p][n];}return num[p];}public static void main(String[] args) {Scanner in = new Scanner(System.in);while (true) {String s = in.nextLine();if (s.equals("")) {break;}//空行单词表结束char[]chs=s.toCharArray();insert(chs);}while (in.hasNext()) {String query = in.nextLine();char[]chs=query.toCharArray();System.out.println(find(chs));}}}