hdu1251 统计难题
统计难题
Time Limit: 4000/2000 MS (Java/Others) Memory Limit: 131070/65535 K (Java/Others)
Total Submission(s): 36652 Accepted Submission(s):
13632
Problem Description
Ignatius最近遇到一个难题,老师交给他很多单词(只有小写字母组成,不会有重复的单词出现),现在老师要他统计出以某个字符串为前缀的单词数量(单词本身也是自己的前缀).
Input
输入数据的第一部分是一张单词表,每行一个单词,单词的长度不超过10,它们代表的是老师交给Ignatius统计的单词,一个空行代表单词表的结束.第二部分是一连串的提问,每行一个提问,每个提问都是一个字符串.
注意:本题只有一组测试数据,处理到文件结束.
注意:本题只有一组测试数据,处理到文件结束.
Output
对于每个提问,给出以该字符串为前缀的单词的数量.
Sample Input
banana
band
bee
absolute
acm
ba
b
band
abc
Sample Output
2
3
1
0
Author
Ignatius.L
Recommend
![](https://images.cnblogs.com/OutliningIndicators/ContractedBlock.gif)
//字典树 人生指针第一份代码 #include<iostream> #include<cstdio> #include<cstring> using namespace std; struct trie { int cnt; trie *next[26]; int flag; }*root; trie *cr() { trie *pt=new trie; pt->cnt=0; memset(pt->next,0,sizeof pt->next); pt->flag=0; return pt; } void insert(char *word) { trie *p=root; char *q=word; while(*q) { int id=*q-'a'; if(p->next[id]==NULL) p->next[id]=cr(); p=p->next[id]; ++q; p->cnt++; } p->flag=1; } int search(char *word) { trie *p=root; char *q=word; while(*q) { int id=*q-'a'; p=p->next[id]; q++; if(p==NULL) return 0; } return p->cnt; } int main() { root=cr(); char s[12]; bool bz=0; while(gets(s)) { if(bz)printf("%d\n",search(s)); else { if(strlen(s)!=0) insert(s); else bz=1; } } return 0; }
折花枝,恨花枝,准拟花开人共卮,开时人去时。
怕相思,已相思,轮到相思没处辞,眉间露一丝。