acwing.142.前缀统计

给定 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>
using namespace std;

const int N=1000010;

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 sum=0;
    int p=0;
  
    for(int i=0;str[i];i++)
    {
        int u=str[i]-'a';
      if(!son[p][u]) break;
        
        p=son[p][u];
     sum+=cnt[p];
    }
    return sum;
}


int main()
{
    int n,m;
    scanf("%d %d",&n,&m);
    while(n--)
    {
       scanf("%s",str);
       insert(str);
    }
    while(m--)
    {
        scanf("%s",str);
        printf("%d\n",query(str));
        
    }
}