HDU-1251统计难题（字典树）

统计难题 Problem Description Ignatius最近遇到一个难题,老师交给他很多单词(只有小写字母组成,不会有重复的单词出现),现在老师要他统计出以某个字符串为前缀的单词数量(单词本身也是自己的前缀).   Input 输入数据的第一部分是一张单词表,每行一个单词,单词的长度不超过10,它们代表的是老师交给Ignatius统计的单词,一个空行代表单词表的结束.第二部分是一连串的提问,每行一个提问,每个提问都是一个字符串. 注意:本题只有一组测试数据,处理到文件结束.   Output 对于每个提问,给出以该字符串为前缀的单词的数量.   Sample Input bananabandbeeabsoluteacmbabbandabc   Sample Output 2310 Author Ignatius.L 我的碎碎念： 上次周赛做到有个题目，是有关字典树的，因为没学过这个知识，当然最后没做出来，所以看书、博客学习了下。 可能是字典树比较简单，还是怎么的，书和博客上关于字典树的内容并不是非常的详细，感觉看完还是不太懂。 便想通过做题来加深理解。 这是做的第一个字典树的水题。 学习到的： 1.gets(数组名+1) 一般使用gets()是需要在其前面加上getchar()的，但从+1的开始便可以不使用getchar()了。 我反思了下自己觉得这个用法新奇的原因，不过是自己学的太死板了。 2.关于字典树 此代码中的y的大小随着你输入内容分的变化而变化。如，小写字母，26=max(y)，大、小写字母，52=max(y)，数字、大、小写字母，91=max(y)。 此代码中t[x].ch[y]用于记录树中不重复的点的个数。 此代码中的t[x].s便是用于记录重复个数的。 这两个的配合使用刚好完成字典树的功能。 #include<iostream> #include<cstdio> #include<cstring> #include<algorithm> #include<cstdlib> using namespace std; struct node { int s,ch[30]; node() { s=0; memset(ch,-1,sizeof(ch)); } }t[410000]; int tot; char st[210]; void bt(char *a,int root) { int x=root,len=strlen(a+1); for(int i=1;i<=len;i++)//这里从1开始，是因为+1了 { int y=a[i]-'a'+1; if(t[x].ch[y]==-1) { t[x].ch[y]=++tot; //printf("t[x].ch[y]=%d\n",t[x].ch[y]); } x=t[x].ch[y]; t[x].s++;//被自己蠢死，t[x].s是用来记录树到某个字母为止，有多少个相同的。 printf("x=%d,tx[x].s=%d\n",x,t[x].s); } } int find1(int root) { int x=root,len=strlen(st+1); for(int i=1;i<=len;i++) { int y=st[i]-'a'+1; if(t[x].ch[y]==-1) return -1; x=t[x].ch[y]; } return x; } int main() { tot=0; while(gets(st+1))//这个非常的机智啊，一般使用gets()均需要使用到getchar(),但从gets(数组名+1)开始就不需要使用了。 { if(st[1]=='\0') break; bt(st,0); } while(gets(st+1)) { int ans=find1(0); if(ans==-1) ans=0; else ans=t[ans].s; printf("%d\n",ans); } return 0; }