将模式串进行建树,每个字符出现的时候都进行标记,即red[p]++;全部建立完毕后
对于每次询问,如果该字符串的某个字符没有出现过,直接结束,不满足该字符串为模式串的前缀,如果出现过,那么p值仍然等于trie[p][x],直到遍历到最后一个字符位置时,返回最后一个字符出现的次数,即为前缀个数。
int find(string s)
{
int p=0;
int sum=0;
for(int i=0;i<s.size();i++)
{
int x=s[i]-'0';
if(!trie[p][x])
return 0;
p=trie[p][x];
}
return red[p];
}
由于对于每次询问,p的值都是从0开始,例如在查找fusu满足多少个字符串的前缀时,akakefusu并不满足,因为fusu中的最后一个字符u与akakefusu中的u的位置不同,也即p的值不同,所以这个不满足前缀。
用memset会MLE
for(int i=0;i<=id;i++)
{
for(int j=0;j<=100;j++)
{
trie[i][j]=0;
}
}
for(int i=0;i<=id;i++)
red[i]=0;
代码:
#include<iostream>
#include<cstring>
using namespace std;
int trie[3000005][100];
int red[3000005];
int id;
int find(string s)
{
int p=0;
int sum=0;
for(int i=0;i<s.size();i++)
{
int x=s[i]-'0';
if(!trie[p][x])
return 0;
p=trie[p][x];
}
return red[p];
}
void insert(string s)
{
int p=0;
for(int i=0;i<s.size();i++)
{
int x=s[i]-'0';
if(trie[p][x]==0)
trie[p][x]=++id;
p=trie[p][x];//记录每个字符出现在值p的位置
red[p]++; //记录每个字符出现在相同位置时的次数
}
}
int main()
{
int t;
cin>>t;
while(t--)
{
int n,m;
cin>>n>>m;
for(int i=0;i<=id;i++)
{
for(int j=0;j<=100;j++)
{
trie[i][j]=0;
}
}
for(int i=0;i<=id;i++)
red[i]=0;
string s;
id=0;
for(int i=0;i<n;i++)
{
cin>>s;
insert(s);
}
for(int i=0;i<m;i++)
{
cin>>s;
cout<<find(s)<<endl;
}
}
return 0;
}