描述
给定N个字符串S1,S2…SN,接下来进行M次询问,每次询问给定一个字符串T,求S1~SN中有多少个字符串是T的前缀。输入字符串的总长度不超过10^6,仅包含小写字母。
输入格式
第一行两个整数N,M。接下来N行每行一个字符串Si。接下来M行每行一个字符串表示询问。
输出格式
对于每个询问,输出一个整数表示答案
样例输入
3 2
ab
bc
abc
abc
efg
样例输出
2
0
字典树
记录每个结点是多少个字符串的末尾结点,查找时累加每个点的数量就可以了
#include <cstdio>
#include <cstring>
using namespace std;
const int maxn = 1000005;
int tot = 1, trie[maxn][26], cnt[maxn];
char s[maxn];
void insert(char *str)
{
int len = strlen(str), p = 1;
for (int k = 0; k < len; ++k)
{
int ch = str[k] - 'a';
if (trie[p][ch] == 0)
trie[p][ch] = ++tot;
p = trie[p][ch];
}
++cnt[p]; //出现数量加1
}
int check(char *str)
{
int len = strlen(str), p = 1, ans = 0;
for (int k = 0; k < len; ++k)
{
p = trie[p][str[k] - 'a']; //下一个节点
if (p == 0)
return ans;
ans += cnt[p]; //加上这个点的字符串数量
}
return ans;
}
int main()
{
int n, m;
scanf("%d%d", &n, &m);
for (int i = 0; i < n; ++i)
{
scanf("%s", s);
insert(s);
}
for (int i = 0; i < m; ++i)
{
scanf("%s", s);
printf("%d\n", check(s));
}
return 0;
}