835. 字符串统计（Trie树模板题）

Posted 2020-11-25 qdu-lkc

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

1. “I x”向集合中插入一个字符串x；
2. “Q x”询问一个字符串在集合中出现了多少次。

共有N个操作，输入的字符串总长度不超过 105105，字符串仅包含小写英文字母。

输入格式

第一行包含整数N，表示操作数。

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

输出格式

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

每个结果占一行。

数据范围

1≤N≤2∗1041≤N≤2∗104

输入样例：

5
I abc
Q abc
Q ab
I ab
Q ab

输出样例：

1
0
1

 

import java.util.Scanner;

public class Main{
        static final int max=100005;
        static int son[][]=new int[max][26];//son数组存储树节点；
        static int cnt[]=new int[max];      //cnt[i]存储以下标i为结点的字符串的数量；
        static int idx;                     //idx表示当前用到了那个下标,下标是0的点既是根节点又是空节点
        static void insert(String s){
                int p=0;
                for(int i=0;i<s.length();i++){
                        int u=s.charAt(i)-‘a‘;
                        if(son[p][u]==0) son[p][u]=++idx;
                        p=son[p][u];
                }
                cnt[p]++;
        }
        static int query(String s){
                int p=0;
                for(int i=0;i<s.length();i++){
                        int u=s.charAt(i)-‘a‘;
                        if(son[p][u]==0) return 0;
                        p=son[p][u];
                }
                return cnt[p];
        }
         public static void main(String[] args) {
                 Scanner scan=new Scanner(System.in);
                 int n=scan.nextInt();
                 while(n-->0){
                         String s=scan.next();
                         if(s.equals("I")){
                                 String str=scan.next();
                                 insert(str);
                         }
                         else{
                                 String str=scan.next();
                                 System.out.println(query(str));
                         }
                 }
        }
}