洛谷真题字典树
P8306 【模板】字典树
1 #include <bits/stdc++.h> 2 using namespace std; 3 int t,n,q; 4 const int maxn=3000005; 5 char s[maxn]; 6 int son[maxn][80],cnt[maxn],idx; 7 int change(char s){ 8 //注意区分大小写字母以及数字 9 if(s>='A'&&s<='Z')return s-'A'; 10 else if(s>='a'&&s<='z')return s-'a'+26; 11 return s-'0'+52; 12 } 13 void insert(char s[]){ 14 int p=0; 15 for(int i=0;s[i];i++){ 16 int u=change(s[i]); 17 if(!son[p][u])son[p][u]=++idx; 18 p=son[p][u]; 19 cnt[p]++; 20 //此题求前缀,并非一模一样的字符串,cnt[p]++要放在for循环里 21 //凡是出现过此字符串都计数,不必等到结束 22 } 23 } 24 int find(char s[]){ 25 int p=0; 26 for(int i=0;s[i];i++){ 27 int u=change(s[i]); 28 if(!son[p][u])return 0; 29 p=son[p][u]; 30 } 31 return cnt[p]; 32 //返回出现次数而非是否出现过 33 } 34 int main(){ 35 scanf("%d",&t); 36 while(t--){ 37 // fill(son[0],son[0]+maxn*80,0); 38 // memset(cnt,0,sizeof cnt); 39 for(int i=0;i<=idx;i++){ 40 //不需要将整个数组用fill归零,时间复杂度会爆(3个点 41 //idx记录了改变过几列的值,只需要将操作过的值归零即可 !!! 42 for(int j=0;j<=80;j++)son[i][j]=0; 43 } 44 for(int i=0;i<=idx;i++)cnt[i]=0; 45 //同son数组,不需要用memset赋值 46 idx=0; 47 //切记idx归零!!! 48 scanf("%d%d",&n,&q); 49 for(int i=1;i<=n;i++){ 50 cin>>s; 51 insert(s); 52 } 53 while(q--){ 54 cin>>s; 55 printf("%d\n",find(s)); 56 } 57 } 58 return 0; 59 }