ACM_模板_字典树

字典树Trie是一门比较简单也比较好懂的算法，因为字典本身是生活中使用率较高的工具，字典的工作原理大家也都懂，其本质也就是前缀查询，对前缀查询的方法，大家很容易可以想到数据结构中的链表结构，用指针去实现一个连接作用。在这里，小编先讲解一下数的指针建立。

首先因为我们字母只有26个，每个节点的指向可以加一个26节点的指针(不考虑大小写)，在每输入一个字符串的时候就建立节点。

小编先给出指针实现的字典树模板：

#include <iostream>//指针实现
#include <cstdio>
#include <cstring>
using namespace std;
char str[11];
struct node
{
	node *next[26];
	int cnt;
	node()
	{//构造函数，创建结点时自动执行
		cnt=0;
		for(int i=0; i<26; i++)
			next[i] = NULL;//将该节点的下面的26个节点初始化为空
	}
};
void insert(node *p,char *str)
{
	for(int i=0; str[i]; i++)
	{
		int t=str[i]-'a';
		if(p->next[t] == NULL)
			p->next[t] = new node;//如果节点不存在，那么先new一个
		p = p->next[t];
		p->cnt++;
	}
}
int find(node *p,char *str)
{
	for(int i=0; str[i]; i++)
	{
		int t=str[i]-'a';
		p = p->next[t];
		if(!p) return 0;
	}
	return p->cnt;
}
int main()
{
	node *root=new node();
	while(gets(str)&&strlen(str))
		insert(root,str);
	while(gets(str))
	{
		int ans=find(root,str);
		printf("%d\n",ans);
	}
	return 0;
}

指针由于使用时，速度会很慢，所以我们在使用的时候有时会很可能超时(TLE)，所以我们可以云耀到数组去模拟指针地址。

小编这里额给出小编的数组模拟字典树Trie模板。

#include <cstdio>//数组模拟实现 
#include <cstring>
#define N 5000000
int tree[N][26],cnt[N];
char str[20];
int tot;
void insertTrie(char *str)
{
	int len=strlen(str);
	int now=0;
	for(int i=0; i<len; i++)
	{
		int w=str[i]-'a';
		if(tree[now][w] == 0) tree[now][w] = ++tot;
		now = tree[now][w];
		cnt[now]++;
	}
}
int searchTrie(char *str)
{
	int len=strlen(str);
	int now=0;
	for(int i=0; i<len; i++)
	{
		int w=str[i]-'a';
		if(tree[now][w] == 0) return 0;
		now = tree[now][w];
	}
	return cnt[now];
}
int main()
{
	while(gets(str)&&strcmp(str,"")!=0)
		insertTrie(str);
	while(gets(str))
		printf("%d\n",searchTrie(str));
	return 0;
}