hdu1251统计难题

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
struct node{
    int count;
    node *next[26]; 
    node(){  //初始化数据 
        memset(next,NULL,sizeof(next));
        count=0;
    }
};
node *p,*root=new node();
void insert(char *s)//插入新单词 
{
    int i,k;
    for(p=root,i=0;s[i];++i)
    {
        k=s[i]-'a';
        if(p->next[k]==NULL) p->next[k]=new node();
        p=p->next[k];
        p->count++;  //记录此字母出现的次数 
    }
}
int search(char *s)
{
    int i,k;
    for(p=root,i=0;s[i];++i)
    {
        k=s[i]-'a';
        if(p->next[k]==NULL) break; //一旦查找不到，立即跳出 
        p=p->next[k];
    }
    if(s[i]) return 0;//s[i]!=0表示中间 
    return p->count;
}
int main()
{
    char s[11];
    while(gets(s),*s) insert(s); //一直读入数据，直到遇到空字符串 
    while(gets(s))
        printf("%d\n",search(s));
    return 0;
}