前缀统计(Trie)

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

算法：
可以用字典树来存储所有的单词，在每一个查询中，从左到右遍历查询字符串，在这一路径上把存在标记的单词个数全部加起来，就是是该查询字符串前缀的字符串个数；

C++代码：

#include <iostream>
#include <cstring>

using namespace std;

const int N = 1e6 + 10;

int n,m;
int son[N][26],cnt[N],idx;
char str[N];

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

int query(char *str){
    int p = 0,res = 0;
    for(int i = 0; str[i]; i++){
        int u = str[i] - 'a';
        if(!son[p][u]) return res;
        p = son[p][u];
        res += cnt[p];
    }
    return res;
}

int main(){
    scanf("%d%d",&n,&m);
    for(int i=  0; i < n; i++){
        scanf("%s",str);
        insert(str);
    }
    for(int i = 0; i < m; i++){
        scanf("%s",str);
        printf("%d\n",query(str));
    }
    return 0;
}