Ignatius最近遇到一个难题,老师交给他很多单词(只有小写字母组成,不会有重复的单词出现),现在老师要他统计出以某个字符串为前缀的单词数量(单词本身也是自己的前缀).
Input
输入数据的第一部分是一张单词表,每行一个单词,单词的长度不超过10,它们代表的是老师交给Ignatius统计的单词,一个空行代表单词表的结束.第二部分是一连串的提问,每行一个提问,每个提问都是一个字符串.
注意:本题只有一组测试数据,处理到文件结束.
Output
对于每个提问,给出以该字符串为前缀的单词的数量.
Sample Input
banana
band
bee
absolute
acm
ba
b
band
abc
Sample Output
2
3
1
0
解题思路:
应该算是字典树的入门题了吧,只要学会关于字典树的最基本操作就可以了。因为我之前没学过,所以在网上看了好久,才一点一点理解了怎么建立怎么搜索。注释写的比较详细,看注释应该可以看懂。
#include<stdio.h>
#include<string.h>
struct trie
{
int count;
trie *next[26];//每个节点的所有子节点包含的字符都不相同,所以只有a-z这26个字母
};
trie root;//根节点不包含字符,作为树的入口
void init()//初始化字典树
{
for(int i=0;i<26;i++)
root.next[i]=NULL;
//没有建立字典树时,根节点没有子节点
}
void insert(char *str)//建立字典树
{
int len=strlen(str);
int num;
trie *head=&root;//对于每一个字符串,都从最根部开始
for(int i=0;i<len;i++)//把字符串中的每一个字符都作为树的一个节点加入树中
{
num=str[i]-'a';
if(head->next[num]==NULL)//树的根节点上还没有num这个子节点
{
head->next[num]=new trie;//在这个子节点上建立一个新树
head->next[num]->count=1;//有1个单词有这个前缀
head=head->next[num];//更新根节点
for(int j=0;j<26;j++)//初始化新树
head->next[j]=NULL;
}
else//已经有num这个子节点了(证明存在有相同前缀的单词)
{
head->next[num]->count++;//有count个单词有从树的根节点到当前节点路径上经过的字母组成的相同前缀
head=head->next[num];//更新根节点
}
}
}
int search(char *str)//查找
{
int len=strlen(str);
int num;
trie *head=&root;//从字典树根部开始查找
for(int i=0;i<len;i++)
{
num=str[i]-'a';
if(head->next[num]==NULL)//节点不存在
return 0;//没有相同前缀的单词,返回
else//节点存在
head=head->next[num];//继续找前缀的下一个字符
}
return head->count;//返回节点的数值,有多少个单词有相同路径(前缀)
}
int main()
{
char word[15];
int ans;
init();
while(gets(word)&&word[0]!='\0')
{
insert(word);//把字符串插入到字典树中
}
while(scanf("%s",word)!=EOF)
{
ans=search(word);//在字典树中找字符串
printf("%d\n",ans);
}
}