Ignatius最近遇到一个难题,老师交给他很多单词(只有小写字母组成,不会有重复的单词出现),现在老师要他统计出以某个字符串为前缀的单词数量(单词本身也是自己的前缀).
注意:本题只有一组测试数据,处理到文件结束.
banana band bee absolute acm ba b band abc
2 3 1 0
这是一道非常经典的字典树题目,赤裸裸的字典树。需要注意的是需要使用C++提交,G++会超。
字典树专题:点击打开链接http://blog.csdn.net/wang_heng199/article/details/76448804
ac代码:
//提交代码使用C++,G++会超内存 #include<iostream> #include<cstdio> #include<cstring> #include<cstdlib> using namespace std; struct node { int count; node *childs[26]; node() { count=0; int i; for(i=0;i<26;i++) childs[i]=NULL; } }; node *root=new node; node *current,*newnode; void insert(char *str)//插入 { int i,m; current=root; for(int i=0;i<strlen(str);i++) { m=str[i]-'a'; if(current->childs[m]==NULL) current->childs[m]=new node; current=current->childs[m]; (current->count)++; } } int search(char *str)//查找 { int i,m; current=root; for(i=0;i<strlen(str);i++) { m=str[i]-'a'; if(current->childs[m]==NULL) return 0; current=current->childs[m]; } return current->count; } int main() { char str[20]; while(gets(str),strcmp(str,"")) insert(str); while(gets(str)!=NULL) printf("%d\n",search(str)); return 0; }
题目链接:点击打开链接http://acm.hdu.edu.cn/showproblem.php?pid=1251