洛谷真题字典树

P8306 【模板】字典树

#include <bits/stdc++.h>
using namespace std;
int t,n,q;
const int maxn=3000005;
char s[maxn];
int son[maxn][80],cnt[maxn],idx;
int change(char s){
//注意区分大小写字母以及数字 
    if(s>='A'&&s<='Z')return s-'A';
    else if(s>='a'&&s<='z')return s-'a'+26;
    return s-'0'+52;
}
void insert(char s[]){
    int p=0;
    for(int i=0;s[i];i++){
        int u=change(s[i]);
        if(!son[p][u])son[p][u]=++idx;
        p=son[p][u];
        cnt[p]++;
        //此题求前缀，并非一模一样的字符串，cnt[p]++要放在for循环里 
        //凡是出现过此字符串都计数，不必等到结束 
    }
}
int find(char s[]){
    int p=0;
    for(int i=0;s[i];i++){
        int u=change(s[i]);
        if(!son[p][u])return 0;
        p=son[p][u];
    }
    return cnt[p];
    //返回出现次数而非是否出现过 
}
int main(){
    scanf("%d",&t);
    while(t--){
//        fill(son[0],son[0]+maxn*80,0);
//        memset(cnt,0,sizeof cnt);
        for(int i=0;i<=idx;i++){
        //不需要将整个数组用fill归零，时间复杂度会爆（3个点
        //idx记录了改变过几列的值，只需要将操作过的值归零即可 ！！！ 
            for(int j=0;j<=80;j++)son[i][j]=0;
        }
        for(int i=0;i<=idx;i++)cnt[i]=0;
        //同son数组，不需要用memset赋值 
        idx=0;
        //切记idx归零！！！ 
        scanf("%d%d",&n,&q);
        for(int i=1;i<=n;i++){
            cin>>s;
            insert(s);
        }
        while(q--){
            cin>>s;
            printf("%d\n",find(s));
        }
    }
    return 0;
}