Trie:高效的存储和查找字符串集合的数据结构
#include <iostream>
using namespace std;
const int N=100010;
int son[N][26], cnt[N], idx; // 下标是0的点,既是根节点又是空结点 cnt:当前下标的值有多少个
char str[N];
void insert(char str[])
{
int p = 0;
for(int i=0;str[i];i++)
{
int u = str[i] - 'a'; //字母编号映射出来
if(!son[p][u]) son[p][u]=++idx;
p = son[p][u];
}
cnt[p]++;
}
int query(char str[])
{
int p = 0;
for(int i=0;str[i];i++)
{
int u = str[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];
cin>>op>>str;
if(op[0] == 'I') insert(str);
else cout<<query(str)<<endl;
}
return 0;
}