hdu 1251 统计难题

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

#include<stdio.h>
#include<math.h>
#include<string.h>
#include<stdlib.h>
#include<iostream>
using namespace std;

struct tree{//建立字典树
    int n;
    tree *next[26];//一个节点拥有26个指针分支节点，有字母时才分配空间
};
tree *root;//建根

void insert(char *p)
{
    tree *now,*cur;
    cur=root;
    int len=strlen(p);
    for(int i=0;i<len;i++)
    {
        if(cur->next[p[i]-'a']==0)//判断该节点的p[i]分支是否开辟空间
        {
            now=(struct tree*)malloc(sizeof(struct tree));
            now->n=1;
            for(int j=0;j<26;j++)
            {
                now->next[j]=0;
            }
            cur->next[p[i]-'a']=now;//指向空间
            cur=cur->next[p[i]-'a'];//指针下移
        }
        else
        {
            cur=cur->next[p[i]-'a'];//已有前缀，指针下移
            cur->n++;
        }
    }
}

int find(char *p)
{
    tree *cur;
    cur=root;
    int len=strlen(p);
    for(int i=0;i<len;i++)
    {
        if(cur->next[p[i]-'a']==0)
        {
            return 0;
        }
        else
        cur=cur->next[p[i]-'a'];//存在前缀，指针下移
    }
    return cur->n;
}

int main()
{
    //freopen("in.txt","r",stdin);
    char p[15];
    root = (struct tree*)malloc(sizeof(struct tree));//开辟空间
    for(int i=0;i<26;i++)
    {
        root->next[i]=0;//初始化0
    }
    root->n=0;

    while(gets(p))
    {
        if(strlen(p)==0)
        break;
        insert(p);
    }
    while(~scanf("%s",p))
    {
        int n=find(p);
        printf("%d\n",n);
    }
    return 0;
}