835. 字符串統計（Trie樹模板題）

維護一個字符串集合，支持兩種操做：

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));
                         }
                 }
        }
}