Trie 142. 前缀统计(acwing)

给定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 <bits/stdc++.h>

using namespace std;
const int N=1e6+5;
int son[N][26],tail[N];//tail某个单词结尾数量
int n,m;
int tot=0;
string s;

void Insert()
{
    int now=0;//root
    for(int i=0;i<s.size();i++)
    {
        int x=s[i]-'a';
        if(son[now][x]==0)
            son[now][x]=++tot;
        now=son[now][x];
    }
    tail[now]++;
}
int query()
{
    int now=0,res=0;
    for(int i=0;i<s.size();i++)
    {
        int x=s[i]-'a';
        now=son[now][x];
        if(now==0)
            return res;
        res+=tail[now];
    }
    return res;
}
int main()
{
    cin >> n >> m;
    while(n--)
    {
        cin >> s;
        Insert();
    }
    while(m--)
    {
        cin >> s;
        cout << query() << endl;
    }
    return 0;
}