字典树模板

#include<stdio.h>
#include<iostream>
#include<string>
#include<cstring>
using namespace std;
//字符数
#define MAX 26
//内存最大限制
#define MEMORYLIMIT 60000*1024
const char CH='a';
struct Node
{
int nCount;
Node * Next[MAX];
};
//先分配好内存，malloc比较浪费时间
Node Memory[MEMORYLIMIT/sizeof(Node)];
struct Trie
{
Node * root;
int allocp;//当前在分配内存中的位置

Trie(){
init();
}
void init(){
allocp=0;
memset(Memory,0,sizeof(Memory));
root=getNode();
}
//从分配内存中获得一个节点
Node * getNode()
{
Node * tmp=&Memory[allocp++];
tmp->nCount=1;
for(int i=0;i<MAX;i++)
tmp->Next[i]=NULL;
return tmp;
}

void insert(char *str)
{
Node *tmp=root;
for(int i=0;str[i];i++)
{
int id=str[i]-CH;
if(tmp->Next[id])
tmp->Next[id]->nCount++;
else
tmp->Next[id]=getNode();
tmp=tmp->Next[id];
}
}

int query(char *str)
{
if(root==NULL)
return 0;
Node *tmp=root;
for(int i=0;str[i];i++)
{
int id=str[i]-CH;
if(tmp->Next[id])
tmp=tmp->Next[id];
else return 0;
}
return tmp->nCount;
}
};
int main()
{

int n,m;
char str[100];
Trie trie;
while(gets(str),*str)
trie.insert(str);
while(gets(str))
printf("%d\n",trie.query(str));
return 0;
}