字典树模板

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#define MAX 26

//Trie的数据结构定义
typedef struct Trie
{
Trie *next[MAX];
int v;   //根据需要变化
};

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]-'0';
if(p->next[id] == NULL)
{
q = (Trie *)malloc(sizeof(Trie));
q->v = 1;    //初始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];
}
}
// p->v = -1;   //若为结尾，则将v改成-1表示(视情况而定)
}

//查找
int findTrie(char *str)
{
int len = strlen(str);
Trie *p = root;
for(int i=0; i<len; ++i)
{
int id = str[i]-'0';  //根据需要选择是减去'0'还是'a'，或者是'A'
p = p->next[id];
if(p == NULL)   //若为空集，表示不存以此为前缀的串
return 0;
if(p->v == -1)   //字符集中已有串是此串的前缀
return -1;
}
//return p->v //返回前缀个数
return -1;   //此串是字符集中某串的前缀
}

//释放内存
int dealTrie(Trie* T)
{
int i;
if(T==NULL)
return 0;
for(i=0;i<MAX;i++)
{
if(T->next[i]!=NULL)
deal(T->next[i]);
}
free(T);
return 0;
}

int main()
{
char str[15];
int i;
for(i=0;i<26;i++)
root.next[i]=NULL;
memset(str,0,sizeof(str));
......
return 0;
}

 

next是表示每层有多少种类的数，如果只是小写字母，则26即可，若改为大小写字母，则是52，若再加上数字，则是62了，这里根据题意来确定。

v可以表示一个字典树到此有多少相同前缀的数目，这里根据需要应当学会自由变化。

Trie的查找（最主要的操作）：
(1) 每次从根结点开始一次搜索；

(2) 取得要查找关键词的第一个字母，并根据该字母选择对应的子树并转到该子树继续进行检索；

(3) 在相应的子树上，取得要查找关键词的第二个字母,并进一步选择对应的子树进行检索。 　　

(4) 迭代过程…… 　　

(5) 在某个结点处，关键词的所有字母已被取出，则读取附在该结点上的信息，即完成查找。