字典树...(eg.hdu 1251)

hdu题目链接

(可以用简单的字符串操作写,更简单…这里是为了学一下字典树的知识点)>_<

---

又称单词查找树，Trie树，是一种树形结构，是一种哈希树的变种。典型应用是用于统计，排序和保存大量的字符串（但不仅限于字符串），所以经常被搜索引擎系统用于文本词频统计。它的优点是：利用字符串的公共前缀来节约存储空间，最大限度地减少无谓的字符串比较，查询效率比哈希表高。

---

字典树与字典很相似,当你要查一个单词是不是在字典树中,首先看单词的第一个字母是不是在字典的第一层,如果不在,说明字典树里没有该单词,如果在就在该字母的孩子节点里找是不是有单词的第二个字母,没有说明没有该单词,有的话用同样的方法继续查找.字典树不仅可以用来储存字母,也可以储存数字等其它数据。

---

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#define Max 26
typedef struct Trie
{
	Trie *next[Max];
	int v;      //eg.上图第一个节点a->v=6,第二层b->v=2;
};
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=(Trie*)malloc(sizeof(Trie));
			q->v=1;
			for(int j=0;j<Max;++j)
			q->next[j]=NULL;
			p->next[id]=q;
			p=p->next[id];
		}
		else
		{
			p->next[id]->v++;
			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->v; 
}
int dealTrie(Trie*T)
{
	int i;
	if(T==NULL) return 0;
	for(i=0;i<Max;i++)
	{
		if(T->next[i]!=NULL)
			dealTrie(T->next[i]);
	}
	free(T);
	return 0;
} 
int main()
{
	root=(Trie*)malloc(sizeof(Trie));
	for(int i=0;i<Max;i++)
	root->next[i]=NULL;
	char str[11];char c;
	while(1) 
	{
		int i;
		char a=getchar();
		if(a!='\n'){
			str[0]=a;
			i=1;
			while((c=getchar())!='\n')
		str[i++]=c;
		str[i]='\0';
		}
		else
		break;
		CreateTrie(str);
	}
	while(scanf("%s",str)!=EOF)
	{
		printf("%d\n",findTrie(str));
	}
	return 0;
}