hdu1251-统计难题-字典树模板题

题目：

Ignatius最近遇到一个难题,老师交给他很多单词(只有小写字母组成,不会有重复的单词出现),现在老师要他统计出以某个字符串为前缀的单词数量(单词本身也是自己的前缀).

思路：

标准的字典树模板

代码：

#include<cstring>
#include<cmath>
#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
struct node{
     int v;
     node *next[26];
     node(){
         memset(next,NULL,sizeof(next));
         v=0;
     }
};
node *p,*root=new node();
void createTrie(char *s)
{
     int i,k;
     for(p=root,i=0;s[i];++i)
     {
         k=s[i]-'a';
         if(p->next[k]==NULL) p->next[k]=new node();
         p=p->next[k];
         p->v++;
     }
}
int findTrie(char *s)
{
     int i,k;
     for(p=root,i=0;s[i];++i)
     {
         k=s[i]-'a';
         if(p->next[k]==NULL) break;
         p=p->next[k];
     }
     if(s[i]) return 0;
     return p->v;
}
int main()
{
     char s[11];
     while(gets(s),*s) createTrie(s);
     while(gets(s))
         printf("%d\n",findTrie(s));
     return 0;
}