题解0013：字典树模板（洛谷）

题目链接：https://www.luogu.com.cn/problem/P8306

题目描述：给定 n 个模式串和 m 次询问，每次询问给定一个文本串，请回答这个文本串是多少个模式串的前缀。

题目思路：

如图，我们可以建立一个这样的树，并从里面查出字符串，这就是字典树。

（字符串要在线上而不是根节点）

当然，这里面的字符要用数字代替，同一个字符编码相同。

我们还要给每个根节点赋值，就是输入字符的顺序。

上图就是依次将 come、time、title、cat、term、can这几个字符存入到字典树的图像。

上代码：

#include<bits/stdc++.h>
using namespace std;
int trie[3000001][65],sum[3000001],n,m,tot=0;//trie的第二个数范围不能太大，可能报错 
string a;
int getnum(char x){//压缩ascal码 
    if(x>='A'&&x<='Z')
        return x-'A';
    else if(x>='a'&&x<='z')
        return x-'a'+26;
    else
        return x-'0'+52;
} 
void Q(){//将字符串存入字典树 
    int as=a.size(); 
    int root=0;//根节点 
    for(int i=0;i<as;i++){
        int id=getnum(a[i]);//字符串的线 
        if(!trie[root][id]){
            trie[root][id]=++tot;//如果前面没数，插入 
        }
        root=trie[root][id];//根节点下移 
        sum[root]++;//存到某节点的前缀数++（递推思路） 
    }
}
int P(){//查询 
    int as=a.size(); 
    int root=0;
    for(int i=0;i<as;i++){
        int id=getnum(a[i]);
        if(!trie[root][id]){//如果前面没数，直接pass 
            return 0;
        }
        root=trie[root][id]; 
    }
    return sum[root];//如果找到，直接返回该点前缀数  
}
int main(){
    int t;
    cin>>t;
    while(t--){
        for(int i=0;i<=tot;i++){
            for(int j=0;j<=122;j++){
                trie[i][j]=0;
            }
        }
        for(int i=0;i<=tot;i++){
            sum[i]=0;
        }//用for循环卡数值清空数组，如果用memset可能空间超限 
        tot=0;
        cin>>n>>m;
        while(n--){
            cin>>a;
            Q();
        }
        while(m--){
            cin>>a;
            cout<<P()<<endl;
        }
    }    
}

完美结束！