静态字典树和动态字典树模板

以前做字典树一直用动态的字典树，每次开辟新空间，需要 new ，删除时delete， 这样动态，节省内存，却要每次开辟和释放空间，饿很难浪费时间，
静态字典树， 是事先开辟一块足够的空间，需要时区取，计较节省时间，有时候却浪费了不必要的内存，写静态字典树，可以开二维数组，也可以用结构体加上指针；

结构体 静态字典树模板

#define  MAX    26 //字符集大小

typedef struct TrieNode
{
int nCount; //记录该字符出现次数
struct TrieNode *next[MAX];
}TrieNode;

TrieNode Memory[1000000];
int allocp = 0;

/*初始化*/
void InitTrieRoot(TrieNode **pRoot)
{
*pRoot = NULL;
}

/*创建新结点*/
TrieNode *CreateTrieNode()
{
int i;
TrieNode *p;

p = &Memory[allocp++];
p->nCount = 1;
for(i = 0 ; i < MAX ; i++)
{
p->next[i] = NULL;
}

return p;
}

/*插入*/
void InsertTrie(TrieNode **pRoot , char *s)
{
int i , k;
TrieNode *p;

if(!(p = *pRoot))
{
p = *pRoot = CreateTrieNode();
}
i = 0;
while(s[i])
{
k = s[i++] - 'a'; //确定branch
if(p->next[k])
p->next[k]->nCount++;
else
p->next[k] = CreateTrieNode();
p = p->next[k];
}
}

//查找
int SearchTrie(TrieNode **pRoot , char *s)
{
TrieNode *p;
int i , k;

if(!(p = *pRoot))
{
return 0;
}
i = 0;
while(s[i])
{
k = s[i++] - 'a';
if(p->next[k] == NULL)    return 0;
p = p->next[k];
}
return p->nCount;
}

动态字典树模板

//专题地址http://vjudge.net/vjudge/contest/view.action?cid=51317#overview 此为动态字典树，静态字典树可自学。
#include <iostream>
#include <cstdio>
#include <string.h>
using namespace std;
typedef struct DT
{
DT *chid[52];
int end;
DT()
{
end=0;
memset(chid,0,sizeof(chid));
}
}dt,*pdt;
pdt root;//定义根节点
int getnum(char s) //根据字符得到num
{
if(s>='A'&&s<='Z')return s-'A';
else return s-'a'+26;
}
void insert(char *s) //插入字符串到字典树
{
pdt next=root;
int num;
for(int i=0;s[i];i++)
{
num=getnum(s[i]);
if(next->chid[num]==NULL) //为空则申请内存
next->chid[num]=new DT(); //开辟内存
next=next->chid[num];
}
next->end=1; //字符串最末尾end赋值true
return ;
}
int query(char *s) //查询字符串在字典树
{
pdt next=root;
int num;
for(int i=0;s[i];i++)
{
num=getnum(s[i]);
if(next->chid[num]==NULL) //没有此地址，返回未找到
return 0;
next=next->chid[num]; //继续查找
}
if(next->end) return 1; //找到返回true
return 0;
}
void del(dt *rot) //递归删除字典树
{
for(int i=0;i<52;i++)
if(rot->chid[i]!=NULL)
del(rot->chid[i]);
delete rot;
}
char str[50002][50];
int main()
{
root=new DT(); //根节点分配内存
char a[]={"nihaoa"};
insert(a);
printf("%d %d\n",query("nihaoa"),query("wobuhao")); //nihaoa存在 返回1 wobuhao不存在 返回0
del(root); //删除字典树
return 0;
}