HDU 1251 统计难题

http://acm.hdu.edu.cn/showproblem.php?pid=1251

第一道Trie树，感觉有点厉害

#include <stdio.h>
#include <string.h> 
#include <stdlib.h> 
const int MAX=26; 
typedef struct Trie
{
    Trie *next[MAX];
    int v;
}Trie;
Trie root;
void create(char *str)
{
    int len=strlen(str);
    Trie *p=&root,*q;
    for(int i=0;i<len;i++)
    {
        int id=str[i]-'a';
        if(p->next[id]==NULL)
        { 
            q=(Trie*)malloc(sizeof(Trie));
            q->v=1;
            for(int j=0;j<MAX;j++)
                q->next[j]=NULL;
            p->next[id]=q;
            p=p->next[id]; 
        }
        else
        {
            p->next[id]->v++;
            p=p->next[id];
        } 
    } 
} 
int find(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 main()
{
    char str[11];
    for(int i=0;i<MAX;i++)
        root.next[i]=NULL; 
    while(gets(str)&&str[0]!='\0')
        create(str);
    while(~scanf("%s",str))
        printf("%d\n",find(str));
    return 0;
}