字典树 前缀树【前缀统计】

给定N个字符串S1,S2…SN，接下来进行M次询问，每次询问给定一个字符串T，求S1～SN中有多少个字符串是T的前缀。

输入字符串的总长度不超过106，仅包含小写字母。

输入格式
第一行输入两个整数N，M。

接下来N行每行输入一个字符串Si。

接下来M行每行一个字符串T用以询问。

输出格式
对于每个询问，输出一个整数表示答案。

每个答案占一行。

输入样例：
3 2
ab
bc
abc
abc
efg
输出样例：
2
0

#include<iostream>
#include<string.h>
using namespace std;

const int N=1000000+10;
int son[N][26],idx,cnt[N];
char str[N];

int n,m;

void insert(){
    int p=0;
    for(int i=0;str[i];i++){
        int s =str[i]-'a';
        if(!son[p][s]) son[p][s] = ++idx;
        p=son[p][s];
    }
    cnt[p]++;
}

int search(){
    int p=0,res=0;
    for(int i=0;str[i];i++){
        int s=str[i]-'a';
        if(!son[p][s]) break;
        p=son[p][s];
        res+=cnt[p];
    }
    return res;
}

int main(){
    ios::sync_with_stdio(false);
    cin>>n>>m;
    for(int i=0;i<n;i++){
        cin>>str;
        insert();
    }
    for(int i=0;i<m;i++){
        cin>>str;
        cout<< search() <<endl;
    }
    return 0;
}