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