HDU 1251 统计难题

    找前缀个数。主要是理解trie树的含义吧，统计一个节点出现的次数就可以了。

#include<stdio.h>
#include<malloc.h>
#include<string.h>
#define MAXNUM 26
typedef struct tnode
{
    int mark,count;
    tnode *next[MAXNUM];
    
}tnode;
tnode *root;
void init()
{
    int i;
    root=(tnode*)malloc(sizeof(tnode));
    for(i=0;i<MAXNUM;i++)
        root->next[i]=NULL;
  root->mark=0;

}
void  insert(char *str)
{
    tnode *r=root,*child;
    int i,flag=0;
    while(*str!='\0')
    {
        if(r->next[*str-'a']==NULL)
        {   flag=1;
            child=(tnode*)malloc(sizeof(tnode));
            for(i=0;i<MAXNUM;i++)
                child->next[i]=NULL;
            child->mark=0;
            child->count=0;
            r->next[*str-'a']=child;


        }
        r->next[*str-'a']->count++;
        r=r->next[*str-'a'];
        str++;

    }

    r->mark=1;

}
void del(tnode *r)
{
    for(int i=0;i<MAXNUM;i++)
    {
        if(r->next[i]!=NULL)
        del(r->next[i]);
    }
    free(r);
}
int search(char *str)
{
    tnode *r=root;
    int i,len;
    len=strlen(str);
    i=0;
    while(r->next[*str-'a']!=NULL)
    {
        if(i==len-1) return r->next[*str-'a']->count;
        r=r->next[*str-'a'];
        str++;
        i++;
    
    }
    return 0;
}

int main()
{
    char str[15];
    int count;
    gets(str);
    init();
    while(strcmp(str,"")!=0)
    {
        insert(str);
        gets(str);
    }
    while(scanf("%s",str)!=EOF)
    {
        count=search(str);
        printf("%d\n",count);

    }
  




    return 0;
}