Hihocoder #1014 : Trie树

输入

输入的第一行为一个正整数n，表示词典的大小，其后n行，每一行一个单词（不保证是英文单词，也有可能是火星文单词哦），单词由不超过10个的小写英文字母组成，可能存在相同的单词，此时应将其视作不同的单词。接下来的一行为一个正整数m，表示小Hi询问的次数，其后m行，每一行一个字符串，该字符串由不超过10个的小写英文字母组成，表示小Hi的一个询问。

在20%的数据中n, m<=10，词典的字母表大小<=2.

在60%的数据中n, m<=1000，词典的字母表大小<=5.

在100%的数据中n, m<=100000，词典的字母表大小<=26.

输出

对于小Hi的每一个询问，输出一个整数Ans,表示词典中以小Hi给出的字符串为前缀的单词的个数。

#include<iostream>
#include<vector>
#include<string>
using namespace std;

struct Trie{
	char key;
	int Count;
	Trie* Next[26];		
}*T;

void buildTree(string str);
void TrieClear(Trie *p);
void InitTrie(int n);

void TrieClear(Trie *p){
	for(int i = 0;i < 26;i++){
		p->Next[i] = NULL;
	}
}

void InitTrie(int n){
	T = (Trie*)malloc(sizeof(Trie));
	T -> Count = 0;
	TrieClear(T);
	for(int i = 0;i< n;i++){
		string str;	cin >> str;
		buildTree(str);				
	}
}

void buildTree(string str){
	Trie* p = T; 
	for(int i = 0;i < str.size(); i++){
		if(p->Next[str[i] - 'a'] == NULL){
			p->Next[str[i] - 'a'] = (Trie*)malloc(sizeof(Trie));
			p = p->Next[str[i] - 'a']; 
			p->key = str[i];
			TrieClear(p);
			p->Count = 1;
		}else{
	 		p = p->Next[str[i] - 'a'];
	 		p->key = str[i];
	 		p->Count += 1;
		}
	}
}

int findCount(string str){
	Trie *p = T;
	for(int i = 0;i < str.size(); i++){
		if(p -> Next[str[i] - 'a'] != NULL){
			p = p -> Next[str[i] - 'a'];
		}else{
			return 0;	
		}
	}
	return p->Count;
}

int main()
{
	int n;
	cin >> n;
	InitTrie(n);
	cin >> n;
	for(int i = 0; i < n;i++){
		string str; cin >> str;
		cout << findCount(str) << endl;;
	}
	return 0;
}