hihocoder1014（trie树）

题目链接

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