142. 前缀统计 ACwing（字典树模板题）

本文主要是介绍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
难度：简单
时/空限制： 1s / 64MB
总通过数： 400
总尝试数： 776
来源：《算法竞赛进阶指南》

思路： 字典树模板题

ACNEW

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>using namespace std;const int maxn = 1e6 + 7;
char s[maxn];
int ch[maxn][26],val[maxn];
int tot = 1;void insert(char *a)
{int u = 1,len = (int)strlen(a);for(int i = 0;i < len;i++){int id = a[i] - 'a';if(!ch[u][id]){ch[u][id] = ++tot;}u = ch[u][id];}val[u]++;
}int query(char *a)
{int ans = 0;int u = 1,len = (int)strlen(a);for(int i = 0;i < len;i++){int id = a[i] - 'a';if(!ch[u][id]){return ans;}u = ch[u][id];ans += val[u];}return ans;
}int main()
{int n,m;scanf("%d%d",&n,&m);for(int i = 1;i <= n;i++){scanf("%s",s);insert(s);}for(int i = 1;i <= m;i++){scanf("%s",s);printf("%d\n",query(s));}return 0;
}

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <stack>
#include <vector>using namespace std;
const int maxn = 1e6 + 7;
int t[maxn][30];
int en[maxn];
char str[maxn];
int n,m,tot;void insert(char *str)
{int len = (int)strlen(str);int p = 1;for(int i = 0;i < len;i++){int ch = str[i] - 'a';if(t[p][ch] == 0)t[p][ch] = ++tot;p = t[p][ch];}en[p]++;
}int search(char *str)
{int len = (int)strlen(str);int p = 1,ans = 0;for(int i = 0;i < len;i++){int ch = str[i] - 'a';p = t[p][ch];if(p == 0)return ans;ans += en[p];}return ans;
}int main()
{int n,m;scanf("%d%d",&n,&m);tot = 1;for(int i = 1;i <= n;i++){scanf("%s",str);insert(str);}for(int i = 1;i <= m;i++){scanf("%s",str);cout << search(str) << endl;}
}