题目链接
http://hihocoder.com/problemset/problem/1014?sid=1355923
题意
先给n个字符串,再给m个字符串,对于m个字符串的每一个,计算之前的n个字符串有多少是以其为前缀的,输出结果。
题解
裸的Trie树,n个字符串先建立Trie树,然后m的每一个都放在m里面跑就行了。
与模板不一样的是,这里需要维护每一个节点被遍历的次数(之前只记录结尾的点),所以用cnt数组来记录。
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <vector>
#include <map>
#include <set>
#include <queue>
using namespace std;
#define INIT(x) memset(x,0,sizeof(x))
#define eps 1e-8
typedef long long ll;
const int inf = 0x3f3f3f3f;
const int maxn = 1000005;
int trie[maxn][26];
int cnt[maxn];
int k = 1;
void insert (char *w) {
int len = strlen(w);
int p = 0;
for(int i=0;i<len;i++) {
int c = w[i]-'a';
if(!trie[p][c]) {
trie[p][c] = k;
k++;
}
p = trie[p][c];
cnt[p]++;
}
}
int search(char *s) {
int len = strlen(s);
int p = 0;
for(int i=0;i<len;i++) {
int c = s[i]-'a';
if(!trie[p][c])
return 0;
p = trie[p][c];
}
return cnt[p];
}
int n,m;
char s[maxn];
int main()
{
cin>>n;
for(int i=0;i<n;i++) {
scanf("%s",s);
insert(s);
}
int ans = 0;
cin>>m;
for(int i=0;i<m;i++) {
scanf("%s",s);
ans = search(s);
cout<<ans<<endl;
}
return 0;
}