AcWing 835. Trie字符串统计
原题链接
简单
作者:
走不到也得走
,
2020-01-19 21:16:58
,
所有人可见
,
阅读 530
#include<iostream>
const int N=100010;
using namespace std;
int son[N][26],cnt[N],idx;
char str[N];
void insert(char s[])
{
int p=0;
for(int i=0;s[i];i++)
{
int u=s[i]-'a';
if(!son[p][u])son[p][u]=++idx;
p=son[p][u];
}
cnt[p]++;
}
int find(char s[])
{
int p=0;
for(int i=0;s[i];i++)
{
int u=s[i]-'a';
if(!son[p][u])return 0;
p=son[p][u];
}
return cnt[p];
}
int main()
{
int n;
scanf("%d",&n);
while(n--)
{
char op[2];
scanf("%s%s",op,str);
if(*op=='I')
{
insert(str);
}
else if(*op=='Q')
{
printf("%d\n",find(str));
}
}
return 0;
}