字典树
把模板背熟
#include <iostream>
using namespace std;
const int N = 100010;
int son[N][26], cnt[N], idx;
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];
scanf("%s%s", op, str);
if (*op == 'I') insert(str);
else printf("%d\n", query(str));
}
return 0;
}
如果涉及到多组数据(多棵树的建立),在每次建立的时候都要记得置零
涉及到不同种类的字符串,要写一个getnum函数
字典树的综合模板
#include <iostream>
using namespace std;
const int N =3e6+10;
int son[3000005][65], cnt[3000005], idx;
char str[3000005];
int getnum(char ch)
{
if(ch>='A'&&ch<='Z')return ch-'A';
else if(ch>='a'&&ch<='z')return ch-'a'+26;
else return ch-'0'+52;
}
void insert(char str[])
{
int p=0;
for(int i=0;str[i];i++)
{
int j=getnum(str[i]);
if(!son[p][j])son[p][j]=++idx;
p=son[p][j];
cnt[p]++;//由于是累计的一个值,所以标记是每次循环都要更新
}
}
int query(char str[])
{
int p=0;
for(int i=0;str[i];i++)
{
int j=getnum(str[i]);
if(!son[p][j])return 0;
p=son[p][j];
}
return cnt[p];
}
int main()
{
int T;
scanf("%d", &T);
while (T--)
{
//要置零
for(int i=0;i<=idx;i++)
for(int j=0;j<=122;j++)
son[i][j]=0;
for(int i=0;i<=idx;i++)
cnt[i]=0;
idx=0;
int a,b;cin>>a>>b;
while(a--)
{
scanf("%s",str);
insert(str);
}
while(b--)
{
scanf("%s",str);
cout<<query(str)<<endl;
}
}
return 0;
}