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

给定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

字典树 前缀树【前缀统计】_第1张图片

#include
#include
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>str;
        insert();
    }
    for(int i=0;i>str;
        cout<< search() <

你可能感兴趣的:(字典树 前缀树【前缀统计】)