前缀统计 Trie

题目链接

https://www.acwing.com/problem/content/144/

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

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

输入格式

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

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

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

输出格式

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

每个答案占一行。

输入样例：

3 2
ab
bc
abc
abc
efg

输出样例：

2
0

题解

字典树算法的板子题 

代码

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn=1e6+5;
char c[maxn];
int son[maxn][26],tot;
int cnt[maxn];
void insert(){
    int len=strlen(c);
    int p=0;  //p为结点在数组里的下标
    for(int i=0;i<len;i++){
        int x=c[i]-'a';
        if(son[p][x]==0) son[p][x]=++tot;
        p=son[p][x];
    }
    cnt[p]++;
}
int search(){
    int len=strlen(c);
    int p=0;
    int ans=0;
    for(int i=0;i<len;i++){
        int x=c[i]-'a';
        if(son[p][x]==0) break;
        p=son[p][x];
        ans+=cnt[p];
    }
    return ans;
}
int main(){
    int n,m;
    cin>>n>>m;
    while(n--){
        scanf("%s",c);
        insert();
    }
    while(m--){
        scanf("%s",c);
        printf("%d\n",search());
    }
    return 0;
}