AcWing 835. Trie字符串统计 c++
原题链接
简单
作者:
wangyj
,
2020-10-15 21:22:44
,
所有人可见
,
阅读 366
#include<iostream>
using namespace std;
int son[100005][26],cnt[100005],idx;
char str[100005];
void insert(char *str)
{
int p=0,i,u;
for(i=0;str[i];i++){
u=str[i]-'a';
if(!son[p][u])son[p][u]=++idx;
p=son[p][u];
}
cnt[p]++;
}
int ask(char *str)
{
int p=0,i,u;
for(i=0;str[i];i++){
u=str[i]-'a';
if(!son[p][u])return 0;
p=son[p][u];
}
return cnt[p];
}
int main()
{
int n;
scanf("%d",&n);
char op[2];
while(n--){
scanf("%s%s",op,str);
if(*op=='I')insert(str);
else printf("%d\n",ask(str));
}
return 0;
}
good day to you! 😊