hdu1251 统计难题（字典树）

#include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <math.h>
#include <algorithm>
using namespace std;
#define maxn 26
typedef struct Trie
{
int v;
Trie *next[maxn];
}Trie;
Trie *root;

void creattrie(char *str)
{
Trie *p=root,*q;
int len=strlen(str);
for(int i=0;i<len;i++)
{
int id=str[i]-'a';
if(p->next[id]==NULL)
{
q=(Trie *)malloc(sizeof(Trie));
for(int j=0;j<26;j++)
q->next[j]=NULL;
q->v=1;
p->next[id]=q;
p=p->next[id];
}
else
{
p->next[id]->v++;
p=p->next[id];

}
}
}

int findtrie(char *str)
{
int len=strlen(str);
Trie *p=root;
for(int i=0;i<len;i++)
{
int id=str[i]-'a';
p=p->next[id];
if(p==NULL)
return 0;
}
return p->v;

}

int deltrie(Trie *t)
{
if(t==NULL)
return 0;
for(int i=0;i<maxn;i++)
{
if(t->next[i]!=NULL)
deltrie(t->next[i]);

}
free(t);
return 0;
}

int main()
{
int i;
char str[11];
root=(Trie *)malloc(sizeof(Trie));
for(i=0;i<maxn;i++)
root->next[i]=NULL;
while(gets(str)&&str[0]!='\0')
creattrie(str);
memset(str,0,sizeof(str));
while(~scanf("%s",str))
{
int ans=findtrie(str);
printf("%d\n",ans);
}
deltrie(root);

return 0;
}