HDU 1251 统计难题【字典树】

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

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

void insert(Trie *p,char *str)
{
    int i=0;
    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=0;i<len;i++)
    {
        int idx=str[i]-'a';
        p=p->next[idx];
        if(p==NULL) return 0;
    }
    return p->cnt;    
}

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