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