AcWing 142 前缀和 字典树

字典树链接：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;
}