HDU 1251 统计难题

#include<iostream>
#include<cstdio>
#include<cstring>
#define maxn 26
using namespace std;
struct node
{
int cnt;
node * next[maxn];
node()
{
cnt=0;
memset(next,0,sizeof(next));
}

};
node *p,*root=new node();
void insert(char *s)
{
p=root;
for(int i=0;s[i];i++)
{
int id=s[i]-'a';
if(p->next[id]==NULL) p->next[id]=new node(); //没有就新建节点
p=p->next[id];//当前节点
p->cnt++;//出现次数加一
}
}
int query(char *s)
{
p=root;
for(int i=0;s[i];i++)
{
int id=s[i]-'a';
if(p->next[id]==NULL) return 0;
p=p->next[id];
}
return p->cnt;
}
int main()
{
char s[20];
while(gets(s),*s) insert(s);
while(gets(s)) cout<<query(s)<<endl;
}