字典树

　　字典树，就是树形的字母排列，从根节点开始(根节点不代表任何字母)，它的每一个子节点都有一个不同的字母，这样，从根节点开始沿任何路径到达一个叶子节点，路过的字母组合就是一个字符串，字典树在字符串的存储和查找上有很大的优越性，下面是用字典树写的两个题。

hdu 1251

/*
这道题就是标准的字典树，存储给出的所有的在字典中给出的字符串和其前缀，统计其数目，存储在字典树中，然后就可以在O(len(m))的时间内查询给出的模式串的答案了
*/
#include<stdio.h>
#include<string.h>
#include<stdlib.h>
struct Trie{
	int v;
    struct Trie *next[26];
} root;
void creat(char *str)
{
	int len = strlen(str);
	struct Trie *p = &root,*q;
	for (int i = 0; i< len; i++){
		int id = str[i] - 'a';
		if (p->next[id] == NULL){
			q = (Trie *)malloc(sizeof(root));
			q->v = 1;
			for (int j = 0; j< 26; j++)
				q->next[j] = NULL;
			p->next[id] = q;
			p = p->next[id];
		}
		else {
			p->next[id]->v++;
			p = p->next[id];
		}
	}
}
int search(char *str)
{
	int len = strlen(str);
	struct Trie *p;
	p = &root;
	for (int i = 0; i< len; i++){
		int id = str[i] - 'a';
		p = p->next[id];
		if (p == NULL) return 0;
	}
	return p->v;
}
int main()
{
	char str[15];
	int i;
	
	for (i = 0; i< 26; i++)
		root.next[i] = NULL;
	while (gets(str) && str[0] != '\0')
		creat(str);
	memset(str,0,sizeof(str));
	while (scanf ("%s",&str) != EOF){
		int ans = search(str);
		printf ("%d\n",ans);
	}
	return 0;
}
pku 1204

这一题我纠结了n久，本来改用AC自动机的(下一篇博客里讲)，因为数据规模的原因，字典树都能暴过，而且由于我的AC自动机学的不怎么的，两个程序跑的时间差不多。。。。

/*
这一题用字典树纯属数据的原因，对所给的模式串建树，再对原图中每个节点的八个方向进行匹配，没有超时很是奇怪
*/
#include<stdio.h>
#include<string.h>
#include<stdlib.h>
struct Trie{
	int count;
	struct Trie *fail;
	struct Trie *next[26];
} *root,*queue[500005];
int n,m,result[1005][2];
char map[1005][1005],ans[1005];
int dir[8][2] = {{-1,0},{-1,1},{0,1},{1,1},{1,0},{1,-1},{0,-1},{-1,-1}};
void insert(char *s,int k)
{
    int len = strlen(s),i;
	struct Trie *p = root,*q;
	for (i = 0; i< len; i++){
		int id = s[i] - 'A';
		if (p->next[id] == NULL){
			q = (struct Trie *)malloc(sizeof(Trie));
			memset(q->next,NULL,sizeof(q->next));
			q->count = -1; p->next[id] = q; p = q;
		}
		else p = p->next[id];
	}
	p->count = k;
}
void search(int x,int y,int k)
{
	int x1 = x,y1 = y;
	struct Trie *p = root;
	while (x1 >= 0 && y1 >= 0 && x1 < n && y1 < m){
		int id = map[x1][y1] - 'A';
		if (p->next[id] == NULL) break;
		else p = p->next[id];
		if (p ->count != -1){
			result[p->count][0] = x; result[p->count][1] = y;
			ans[p->count] = k + 'A'; p->count = -1;
		}
		x1 += dir[k][0]; y1 += dir[k][1];
	}
}
void solve(int sum)
{
	int i,j,k;
	for (i = 0; i< n; i++)
		for (j = 0; j< m; j++)
			for (k = 0; k< 8; k++)
				search(i,j,k);
    for (i = 0; i < sum; i++)
		printf ("%d %d %c\n",result[i][0],result[i][1],ans[i]);

}
int main()
{
	int i,k;
	char word[2005];
	scanf ("%d%d%d",&n,&m,&k);
	memset(map,0,sizeof(map));
	root = (struct Trie *)malloc(sizeof(Trie));
	memset(root->next,NULL,sizeof(root->next));
	for (i = 0; i < n; i++)	scanf ("%s",&map[i]);
	getchar();
	for (i = 0; i< k; i++){
		gets(word);
		insert(word,i);
	}
	solve(k);
	return 0;
}