HDU 1251 统计难题【字典树】

题意：中文题--跟着模板敲的--第一棵字典树--@_@

 #include<iostream>
 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 using namespace std;
 typedef struct node
 {
     int cnt;
     struct node *next[];
 } Trie;
 Trie *create()
 {
     Trie *p=(Trie *)(malloc)(sizeof(Trie));
     p->cnt=;
     for(int i=;i<;i++) p->next[i]=NULL;
     return p;
 }

 void insert(Trie *p,char *str)
 {
     int i=;
     while(str[i])
     {
         int idx=str[i]-'a';
         if(p->next[idx]) p->next[idx]->cnt++;
         else p->next[idx]=create();
         p=p->next[idx];
         i++;
     }
 }
 int search(Trie *p,char *str)
 {
     int len=strlen(str);
     for(int i=;i<len;i++)
     {
         int idx=str[i]-'a';
         p=p->next[idx];
         if(p==NULL) return ;
     }
     return p->cnt;
 }

 int main()
 {
     Trie *p=create();
     char str[];
     while(gets(str)&&str[])
     insert(p,str);
     while(scanf("%s",str)!=EOF)
     printf("%d\n",search(p,str));
 }