模拟Trie树结构

    static int N=100010;
    static int [][]son=new int[N][26];  //son[][]存储树中每个节点的子节点
    static int []cnt=new int[N];        //记录以每个结点结尾的单词数量
    static int idx;         //当前用的的哪个下标，下标0：既是根节点又是空节点
    //插入操作
    public static void insert(char []str){
        int p=0;//根节点
        for (int i = 0; i < str.length; i++) {//从根节点开始依次遍历
            int x=str[i]-'a';//把当前这个节点的下标取出来
            //如果当前这个点上不存在对应的字母的话，创建出来
            if(son[p][x]==0){
              son[p][x]=++idx;
            }
            //走到下一个点,p可以理解为父节点
            p=son[p][x];
        }
        cnt[p]++; //记录下这个单词出现次数
    }
    //查询操作
   public static int query(char []str){
       int p=0;
       for (int i = 0; i < str.length; i++) {
           int x=str[i]-'a';
           //如果不存在这个子节点的话，说明集合中不存在这个单词
           if(son[p][x]==0) return 0;
           p=son[p][x];
       }
       return cnt[p];
   }