HDU-1251 统计难题 （字典树）

题目链接

题意：

统计出以某个字符串为前缀的单词数量(单词本身也是自己的前缀).

输入数据的第一部分是一张单词表,每行一个单词,单词的长度不超过10,它们代表的是老师交给Ignatius统计的单词,一个空行代表单词表的结束.第二部分是一连串的提问,每行一个提问,每个提问都是一个字符串

思路：
字典树，+上一个计数数组统计到该字符有多前缀。

注意：开的数组要大于400000，不然会Timit ，过大会超内存。

AC：

#include<iostream>
#include<cstring>
#include<stdio.h>
#include<math.h>
using namespace std;
const int maxn = 400000;
int Trie[maxn][27], tot = 1;
int Cnt[maxn];
void init(){
	memset(Trie, 0, sizeof(Trie));
	memset(Cnt, 0, sizeof(Cnt));
}
void Insert(char *s){
	int len = strlen(s);
	int u = 1;
	for(int i = 0; i < len; ++i) {
		int c = s[i] - 'a';
		if( !Trie[u][c] ) {
			Trie[u][c] = ++tot;
		}
		u = Trie[u][c];
		Cnt[u]++;
	}
	return ;
}
int find(char *s) {
	int len = strlen(s);
	int u = 1;
	for(int i = 0; i < len; ++i) {
		int c = s[i] - 'a';
		if( !Trie[u][c] ) {
			return 0;
		}
		u = Trie[u][c];
	}
	return Cnt[u];
}

int main() {
	char s[105];
	init();
	while(gets(s)){
		if(s[0] == 0){
			break;
		}
		Insert(s);
	}
	while(scanf("%s", &s) != EOF){
		printf("%d\n", find(s));
	}
	return 0;
}