Trie练习题java--Trie字符串统计

前言：

最近时间有点紧o(╥﹏╥)o，没有时间进行过多的说明，有时间了再补起来，先把学习的东西记录一下，本文仅供个人学习使用，作为学习记录激励自己，当然也希望能在一定程度上帮到其他人，如果有错误还望批评指正。

题：

维护一个字符串集合，支持两种操作：

1. I x 向集合中插入一个字符串 x；
2. Q x 询问一个字符串在集合中出现了多少次。
   共有 N 个操作，输入的字符串总长度不超过 105，字符串仅包含小写英文字母。

输入格式
第一行包含整数 N，表示操作数。

接下来 N 行，每行包含一个操作指令，指令为 I x 或 Q x 中的一种。

输出格式
对于每个询问指令 Q x，都要输出一个整数作为结果，表示 x 在集合中出现的次数。

每个结果占一行。

数据范围
1≤N≤2∗104
输入样例：

5
I abc
Q abc
Q ab
I ab
Q ab

输出样例：

1
0
1

图：

这里先借百度的图片先用用，之后有时间会补上的o(╥﹏╥)o，trie树不知道是啥的朋友们还请先去学习trie数之后再来围观。

代码（含注释解释）

import java.io.*;

/*
    简明一下思路：
        这里用trie树来存放每个单词中的每个字母，用数组来模拟trie，因为速度快，
        这里算一下时间复杂度就知道，我就不算了
        
        思路：
            将每个单词的字母拆开放入到数组模拟的trie树内，
            这里就简单介绍一下，每个节点的确定方式吧：
                这里用到了2维数组，因为要模拟trie树，树结构要用数组模拟的话，至少需要两个维度的
                下面会对两个维度的意义进行说明，每个节点以son[p][u]为例：
                    p: 作为一维坐标，用来表示当前节点所在的一维位置，
                        在trie树生长（简单来说就是在你往数组里面加字母的时候）的时候，通过idx指针来进行开拓(创建新节点)维护。
                    u: 作为二维坐标，用来表示当前节点的字母，并作为节点的标识；
                        这里会对字母做一个映射的转换，
                        比如说a就对应1，b对应2，这个根据自己的喜好来就好，
                        我是直接将字母的ASCII码拿过来用然后减掉a的ASCII码做映射的，好耶~，偷懒是好文明(*^▽^*)。
                    son[p][u]：作为每个节点的指针域，用来指向下一个节点的一维位置，
                        也就是son[p][u] = 下一个节点的p坐标，
                        然后这个p坐标又是通过idx指针来进行维护开拓的，所以在开拓的时候可以等同于son[p][u] = ++idx;
            
                说了那么多是不是还是觉得脑袋萌萌的，其实二维数组你把它先看为一维数组先，然后这个一维数组的每个单元，都是一个数组，
                在这个基础上看，其实这个一维数组的每个单元（也就是（把第一个括号内的东西看成一个一维数组）-->(son[p])[u]）就代表了每个字母,
                虽然不是很准确，但是应该能帮助点理解吧。
            然后每个单词的结尾是需要进行标识的，我们在用一个一维的数组在结尾的一维节点的地方（也就是开拓终点的p的值）记录一下就好
*/

public class Main{
    //son|-->存放<--|该节点的左节点的大致位置，也就是确定该节点的下一个节点的p坐标（第一维度坐标）
    static int[][] son = new int[100010][26];
    //cn数组存放一个单词结尾字符的次数
    static int[] cn = new int[100010];
    //idx指针用来|-->指定<--|下一个节点的大概位置，也就是确定该节点的下一个节点的p坐标
    static int idx;
    public static void main(String[] args) throws IOException{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int n = Integer.parseInt(br.readLine());
        char[] s;
        
        while(n-->0){
            s = br.readLine().toCharArray();
            //把输入的单词放入到模拟的trie树中去
            if(s[0] == 'I'){
                int p = 0;
                //这里从角标2开始遍历，因为前两个角标是字符（'I'/'Q'）和空格
                for(int i = 2;i<s.length;i++){
                    int u = s[i]-'a';       //将字符映射成数组角标，从a~z，映射成0~25，并将映射后的数组角标作为son的第二位坐标
                    //通过p（第一维坐标）和u(第二维坐标)来唯一确定一个节点
                    //判断当前节点有没有下一个节点，没有的话就创建一个新的节点
                    if(son[p][u] == 0) son[p][u] = ++idx;
                    //将p坐标变更为下一个节点的p坐标
                    p = son[p][u];
                }
                //出了循环，代表一个单词已经成功添加进了模拟的trie树中，这个时候记录结果在cn数组里面
                cn[p]++;
            } else if(s[0] == 'Q'){     //查询，思路和上面类似，就不做多余的说明了
                int p = 0;
                int i;
                for(i = 2;i<s.length;i++){
                    int u = s[i]-'a';
                    if(son[p][u] == 0) {
                        System.out.println(0);
                        break;
                    }else p = son[p][u];
                }
                if(i==s.length) System.out.println(cn[p]);
            }
            
        }
        
    }
}