题目大意:给你很多单词,要求查找以某些单词为前缀的单词数量有多少?
解题思路:很明显的一道字典树的入门题。一直都习惯用数组code的,突然看资料都是指针写的,很不习惯呀!不过重新看了下基础,回过头来写,发现还不是那么难,很好写的。言归正传,字典树主要就是那几步,一是建立树,查找和释放。资料在网上有很多的,就不赘述了。code给了很详细的解释了,详见code。
题目来源:http://acm.hdu.edu.cn/showproblem.php?pid=1251
code:
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int MAXN = 26;
typedef struct node{
int cnt; //记录结点出现该词的次数
node* next[26]; //存储结点的下一个字符
}Trie;
Trie root; //字典树的根节点
void createTrie(char* str){
int len=strlen(str); //计算词的长度
Trie *p=&root,*q; //初始化当前结点为根节点,以及中间变量的指针
for(int i=0;i<len;i++){ //将每个字符依次插入字典树中
int id=str[i]-'a'; //计算下标
if(p->next[id]==NULL){ //如果没有出现过,则增加结点
q=new Trie; //初始化一个新节点,开辟一个空间
q->cnt=1; //将其出现次数初始化为1
for(int j=0;j<MAXN;j++) //将其子结点初始化为NULL
q->next[j]=NULL;
p->next[id]=q; //将其付给p的下一个结点
p=p->next[id]; //指针下移
}
else{ //如果出现过
p->next[id]->cnt++; //将对应的子结点的计数加一
p=p->next[id]; //指针下移
}
}
}
int findTrie(char* str){
int len=strlen(str); //计算关键词的长度
Trie *p=&root; //初始化当前结点为根节点
for(int i=0;i<len;i++){ //依次查找
int id=str[i]-'a'; //计算下标
p=p->next[id]; //下移指针查找
if(p==NULL) return 0; //如果没有比较完成就不匹配,则说明该字典树中没有该词
}
return p->cnt; //如果遍历完成,存在则返回叶子结点的计数值
}
int main(){
char str[11];
for(int i=0;i<MAXN;i++)//初始化根节点所有子结点为空
root.next[i]=NULL;
while(gets(str) && str[0]!='\0') //接收词库
createTrie(str); //创建字典树
memset(str,0,sizeof(str)); //将数组置0,接收查询键值
while(scanf("%s",&str)!=EOF){ //接收查询关键字
int ans=findTrie(str);
printf("%d\n",ans);
}
return 0;
}