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

835. Trie字符串统计

题目链接https://www.acwing.com/problem/content/837/
题目：

思路： t=0表示从头结点0开始出发,p[t][a]==0表示a不存在，而t表示在a的前面一个字母。cnt[t]表示以t结尾的字母的个数

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>

using namespace std;
const int N=6e5+10;
char a[N],op[2];
int p[N][30],idx=0;
int cnt[N];
void insertt(){
    int t=0;
    for(int i=0;a[i];i++){
        if(p[t][a[i]-'a']==0){
            p[t][a[i]-'a']=++idx;
        }
         t=p[t][a[i]-'a'];
    }
    cnt[t]++;
}
int query(){
    int t=0;
    for(int i=0;a[i];i++){
        if(p[t][a[i]-'a']==0) return 0;
        t=p[t][a[i]-'a'];
    }
    return cnt[t];
}
int main(){
    int n;
    cin>>n;
    while(n--){
        scanf("%s %s",op,a);
        //cout<<*op<<" "<<a<<endl;
        if(op[0]=='I'){
            insertt();
        }else{
            printf("%d\n",query());
        }
    }
    return 0;
}