字典树链接:https://www.cnblogs.com/fusiwei/p/11972776.html
题目描述
给定N个字符串S1,S2…SNS1,S2…SN,接下来进行M次询问,每次询问给定一个字符串T,求S1~SNS1~SN中有多少个字符串是T的前缀。
输入字符串的总长度不超过106106,仅包含小写字母。
输入格式
第一行输入两个整数NN,MM。
接下来NN行每行输入一个字符串SiSi。
接下来MM行每行一个字符串T用以询问。
输出格式
对于每个询问,输出一个整数表示答案。
每个答案占一行。
样例
输入样例:
3 2
ab
bc
abc
abc
efg
输出样例:
2
0
思路:字典树。把N个字符串插入一棵trie树,trie树上的每个节点储存一个整数cnt,记录该节点是多少个字符串的末尾节点,询问时累加途径的每个节点的cnt值。
代码如下:
#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e6+10;
int n,t,tot=1;
char str[maxn];
int trie[maxn][26],End[maxn];
void insert(char str[]){
int len =strlen(str),p=1;//1.令指针p初始指向根节点
for(int i=0;i<len;i++){
int ch=str[i]-'a';//2.依次扫描str中的每个字符
if(trie[p][ch]==0){//3。若p的c字符指针指向空,就新建一个节点Q,令P的c字符指针指向Q;
trie[p][ch]=++tot;
}
p=trie[p][ch];//令P=Q
}
End[p]++;//当str中的字符扫描完毕时,在当前节点p上标记它是一个字符串的末尾,因为此处可能有重复字符串,所以令为int数组,并++
}
int search(char str[]){
int len=strlen(str),p=1,ans=0;//累加和
for(int i=0;i<len;i++){//把这个字符串所有的前缀和都找出来
p=trie[p][str[i]-'a'];
if(p==0) return ans;
ans+=End[p];
}
return ans;
}
int main(){
cin>>n>>t;
for(int i=0;i<n;i++){
cin>>str;
insert(str);
}
while(t--){
cin>>str;
cout<<search(str)<<endl;
}
return 0;
}