解题思路
知识点:字典树
(嗯嗯,我就是来安利博客的 ^ _^ )
学会字典树的基本操作后,回来看看这道题,啊,模板题。。。
先把
N
N
N个字符串存入字典树,令
v
[
p
]
v[p]
v[p]表示以字典树上节点p为结尾的字符串的个数。
计算答案:我们遍历字符串T,答案为
∑
x
ϵ
p
[
T
]
v
[
p
]
\sum _{x\epsilon p[T]} v[p]
∑xϵp[T]v[p]
(
p
[
T
]
p[T]
p[T]表示T在字典树上遍历过的节点编号集合)
代码
#include<iostream>
#include<cstdio>
#include<iomanip>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
char a[1000010],s[1000010];
int n,m,trie[1000010][30],v[1000010],lyx=1;
void insert(){//往字典树加入一个字符串
int len=strlen(a+1),p=1,c=0;
for(int i=1;i<=len;i++)
{
c=a[i]-'a'+1;
if(!trie[p][c])
trie[p][c]=++lyx;
p=trie[p][c];
}
v[p]++;
// cout<<v[p]<<" "<<p;
}
int get(){//查询
int len=strlen(s+1),p=1,ans=0,c=0;
for(int i=1;i<=len;i++)
{
c=s[i]-'a'+1;
if(!trie[p][c])
return ans;
p=trie[p][c];
ans+=v[p];
}
return ans;
}
int main() {
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
{
scanf("%s",a+1);
insert();
}
for(int i=1;i<=m;i++)
{
scanf("%s",s+1);
printf("%d\n",get());
}
}