hdu 1251 前缀树

题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=1251

    普通trie树，只是在构建的时候统计出每个节点儿子的个数，最后直接匹配每个前缀，输出最后一个前缀字符在树中儿子的个数加一（其本身）即可。

    纠结的是题目没给数据范围，tree数组开100010 RE，开200010 还是RE，一狠心开了500010，果断AC。。。

 

代码：

 

 

#include<iostream>
#include<cstring>
#include<cstdio>
using  namespace std ;
int num ;
int ans ;
struct node{
     int n ;
     int next[ 30] ;
     void init(){
        memset(next, 0, sizeof(next)) ;
        n= 0 ;
    }
}tree[ 500010] ;

void insert( char a[]){
     int index= 0 ;
     int len=strlen(a) ;
     for( int i= 0;i<len;i++){
         if(tree[index].next[a[i]- ' a ']== 0){
            tree[++num].init() ;
            tree[index].next[a[i]- ' a ']=num ;
            index=num ;
            tree[index].n= 1 ;   // 新节点没有儿子，但其本身也为本身的前缀
        }
         else{
            index=tree[index].next[a[i]- ' a '] ;
            tree[index].n ++ ;  // 每匹配一个字符，儿子个数增一
        }
    }
}
int find( char a[]){
     int index= 0 ;
     int len=strlen(a) ;
     if(len== 0)   return  0 ;
     for( int i= 0; i<len; i++){
         if(tree[index].next[a[i]- ' a ']== 0)
             return  0 ;
         else
            index=tree[index].next[a[i]- ' a '] ;
    }
     return tree[index].n ;
}
int main(){
     char str[ 15] ;
     int t= 0 ;
    tree[ 0].init() ;
    num =  0 ;
    ans =  0 ;
     while(gets(str),strcmp(str, "")!= 0)       // 获取以空行区分的字符串
        insert(str) ;
     while(scanf( " %s ", str)!=EOF)
        cout << find(str) << endl ;
     return  0 ;
}