hdu 1251简单字典树

#include<stdio.h>
#include<iostream>
#include<string.h>
using namespace std;
struct node {
int num,i;
node *a[27];
node (){
num=0;
for(i=0;i<27;i++)
a[i]=NULL;
}
}*root;
char str[20];
void insert(int len,node *root) {
int i;
node *q;
for(i=0;i<len;i++) {
if(root->a[str[i]-'a']==NULL) {
q=new node();
q->num++;
root->a[str[i]-'a']=q;
root=q;
}
else  {
root=root->a[str[i]-'a'];
root->num++;
}
}
return ;
}
int find(int len,node *root) {
int i;
for(i=0;i<len;i++) {
root=root->a[str[i]-'a'];
if(root==NULL)
return 0;
}
return root->num;
}
int main() {
int k;
root=new node();
while(gets(str)&&str[0]!='\0') //输入条件太恶心了
insert(strlen(str),root);
while(scanf("%s",str)!=EOF) {
k=find(strlen(str),root);
printf("%d\n",k);
}
return 0;
}