统计难题（字典树的最简单吧）hdu 1251

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

 

就是建立一颗字典树，字典树定义相信看看百度就懂，不过写树不会写了很久。。。发现指针在建树的时候连不上，后来发现是因为我当q=NULL的时候就给他建树，但实际上这时候q已经跟我的树没有了关系，所以他建立的新空间等于没用，改了就好了，注意查找的时候某些是没有的，容易查找到NULL的空间而爆掉。。。

#include<stdio.h>
#include<string.h>
#include<malloc.h>
struct p
{
    int no;
    struct p *son[30];
}*father[30];

int check(char s[30])
{
    struct p *q=father[s[0]-'a'];
    int x,l=strlen(s);
    if(q==NULL)
    {
        printf("0\n");
        return 0;
    }
    for(x=1;x<l;x++)
    {
        q=q->son[s[x]-'a'];
        if(q==NULL)
        { 
            printf("0\n");
            return 0;
        }
    }
    printf("%d\n",q->no);
    return 0;
}   
void set()
{
    int x,y;
for(x=0;x<26;x++)
        father[x]=(struct p*)malloc(sizeof(struct p));
    for(x=0;x<26;x++)
    {
        for(y=0;y<26;y++)
            father[x]->son[y]=NULL;
        father[x]->no=0;
    }
    return ;
}
void insert(int l,char s[30])
{
    int x,y;
        struct p *q=father[s[0]-'a'];
        q->no++;
        for(x=1;x<l;x++)
        {
            if(q->son[s[x]-'a']==NULL)
            {
                q->son[s[x]-'a']=(struct p*)malloc(sizeof(struct p));
                q=q->son[s[x]-'a'];
                for(y=0;y<26;y++)
                    q->son[y]=NULL;
                q->no=0;
            }
            else
                q=q->son[s[x]-'a'];
            q->no++; 
        }
   return ;
}
int main()
{
    char s[15];
    int checkit=0;
 
    set();

    while(gets(s))
    {
        int l=strlen(s);
        if(l==0)
        {
            checkit=1;
            continue;
        }
        if(checkit==0)
            insert(l,s);
        else
            check(s);
    }
    return 0;
}