Problem Description Ignatius最近遇到一个难题,老师交给他很多单词(只有小写字母组成,不会有重复的单词出现),现在老师要他统计出以某个字符串为前缀的单词数量(单词本身也是自己的前缀).
Input 输入数据的第一部分是一张单词表,每行一个单词,单词的长度不超过10,它们代表的是老师交给Ignatius统计的单词,一个空行代表单词表的结束.第二部分是一连串的提问,每行一个提问,每个提问都是一个字符串.
注意:本题只有一组测试数据,处理到文件结束.
思路：
就是构建+查找 构建字典树的时候，注意num 的用处和flag的用处，分别代表着单词的数目和结尾 标志。
#include <iostream> #include <cstdio> #include <cstring> using namespace std; const int base = 'a',maxn = 26; char s1[12],s2[12]; struct Tire { int num; int flag; Tire *son[maxn]; Tire() { //注意初始化 num = 1; flag = false; memset(son,NULL,sizeof(son)); } }; /*-------------构造字典树------------*/ Tire* NewTire() { Tire *temp = new Tire; return temp; } void Insert(Tire *root,char *s) { Tire *temp = root; while(*s) { if(temp->son[*s-base] == NULL) { temp->son[*s-base] = NewTire(); } else temp->son[*s-base]->num++; temp = temp->son[*s-base]; s++; } temp->flag = true; } int Serach(Tire *root,char *s) { Tire *temp = root; while(*s) { if(temp->son[*s-base] == NULL) return 0; temp = temp->son[*s-base]; s++; } return temp->num; } int main() { //freopen("in.txt","r",stdin); Tire *root = NewTire(); root->num = 0; while(gets(s1) && strcmp(s1,"") != 0) { Insert(root,s1); } while(cin>>s2) { int ans = Serach(root,s2); cout<<ans<<endl; } return 0; }