字典树作为字符串中最常见的一种算法,在比赛中也最容易出现,写此篇文章的目的一是给自己做一个模板,二是希望用自己的语言和思路给大家呈现一个比较简单的学习算法的资料,希望对大家有所帮助。(本人菜鸟,望各路大神指导纠正或请大神绕道而走)
定义结构体:
struct Trie
{
Trie *next[26];//用于存放子节点的地址
int v;记录以该节点为根的树叶子节点的个数
};
Trie *root;//根节点,根节点需在主函数当中用root = newnode()开辟空间
生成新结点函数:
Trie *newnode()//初始化每个新节点
{
Trie *t;
t = new Trie;
for(int i = 0; i<26; i++)
t->next[i] = NULL;
return t;
}
创建字典树:
void CreatTrie(char *str)//创建字典树
{
int n = strlen(str);
Trie *q, *p = root;
for(int i = 0; i < n; i++)
{
int id = str[i] - 'a';//得出i节点的编号
if(p->next[id] == NULL)//如果第id个节点为空,则创建新节点
{
q = newnode();
q->v = 1;//初始化为1
p->next[id] = q;
p = p->next[id];
}
else
{
p->next[id]->v++;//如果第id个节点不为空,则v++
p = p->next[id];//指针向下移动
}
}
}
查找函数:
int search(char *str)
{
int m = strlen(str);
Trie *p = root;
for(int i = 0; i < m; i++)
{
int id = str[i] - 'a';
p = p->next[id];//每次都先进行移动,然后进行判断
if(p == NULL) return 0;//若为空,表示不存以此为前缀的串,直接跳出
}
return p->v;//最后返回p结点的v值
}
删除函数:
int dealTrie(Trie* T)//动态删除,防止内存溢出
{
int i;
if(T==NULL) return 0;
for(i=0;i<26;i++)
if(T->next[i]!=NULL) dealTrie(T->next[i]);
delete(T);
return 0;
}
下面给出网上个人认为比较好的一篇:
结构体定义:
#define MAX 26
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';
p = p->next[id];
if(p == NULL) //若为空集,表示不存以此为前缀的串
return 0;
if(p->v == -1) //字符集中已有串是此串的前缀
return -1;
}
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;
}