HDU1251 统计难题 Trie树

题目很水,但毕竟是自己第一道的Trie,所以还是发一下吧.Trie的更多的应用慢慢学,AC自动机什么的也慢慢学....

#include<iostream>
#include<cstring>
#include<string>
#include<cstdio>
#include<algorithm>
#include<cmath>
#define maxn
using namespace std;
 
struct TrieNode
{
	int cnt;
	int flag;
	TrieNode *next[26];
}T[500000],*Trie;
 
int top;
 
void insert(char *c)
{
	int len=strlen(c);TrieNode *p=Trie;
	for(int i=0;i<len;++i){
		if(p->next[c[i]-'a']!=NULL){
			p=p->next[c[i]-'a'];
			p->cnt++;
		}
		else{
			T[top++].cnt=1;
			p->next[c[i]-'a']=&T[top];
			p=p->next[c[i]-'a'];
		}
	}
	p->flag=true;
}
 
int find(char *c)
{
	int len=strlen(c);TrieNode *p=Trie;
	for(int i=0;i<len;++i){
		if(p->next[c[i]-'a']!=NULL){
			p=p->next[c[i]-'a'];
		}
		else{
			return 0;
		}
	}
	return p->cnt;
}
 
int main()
{
	char str[15];top=0;
	Trie=&T[top++];bool flag=false;;
	while(gets(str))
	{
		if(flag){
			printf("%d\n",find(str));
			continue;
		}
		if(strlen(str)==0){
			flag=true;continue;
		}
		insert(str);
	}
	return 0;
}