现在的位置: 首页 > 综合 > 正文

Trie树学习小记 Poj 3630 & Hdu1671 Phone List

2013年04月26日 ⁄ 综合 ⁄ 共 2013字 ⁄ 字号 评论关闭

最后知识总结部分转自 http://www.ahathinking.com/archives/14.html

Poj 3630 & Hdu1671 Phone List

题意:给出一组电话号码,判断其中一个号码是否另一个号码的前缀

#include <cstdio>
#include <cstring>

int n,e;
struct Trie
{
	bool flag;
	int next[11];
}trie[100010];

char str[15];

bool Insert ()
{
	int i,p=1;
	bool f=true,f2=false;
	for (i=0;str[i];i++)
    {		
		if (trie[p].next[str[i]-'0']==0)   //构造新的节点
		{
			trie[p].next[str[i]-'0']=++e;
			f2=true;        //创建了一个新节点
		}
		p=trie[p].next[str[i]-'0'];
		if (trie[p].flag)  //如果这个节点是之前某个单词的结束
			f=false;
    }
	trie[p].flag = true;   //标记该点为一个单词的结束
    if (f2==false)         //如果没有创建新节点
        f=false;
    return f;
}

int main ()
{
#ifdef ONLINE_JUDGE        
#else  
    freopen("read.txt","r",stdin);
#endif
	int T;
	scanf("%d",&T);
	while (T--)
	{
		bool flag=true;
		e=1;
		memset(trie,0,sizeof(trie));
		scanf("%d",&n);
		while (n--)
		{
			scanf("%s",str);
			if (Insert()==false)
				flag=false;
		}
		if (flag)
			printf("YES\n");
		else
			printf("NO\n");
	}
	return 0;
}

/*
2
2
1
12
2
12
1
NO
NO
*/

文章作者:Yx.Ac   文章来源:勇幸|Thinking (http://www.ahathinking.com)   转载请注明,谢谢合作。

    前几天学习了并查集和trie树,这里总结一下trie。

    本文讨论一棵最简单的trie树,基于英文26个字母组成的字符串,讨论插入字符串、判断前缀是否存在、查找字符串等基本操作;至于trie树的删除单个节点实在是少见,故在此不做详解。

  • Trie原理

Trie的核心思想是空间换时间。利用字符串的公共前缀来降低查询时间的开销以达到提高效率的目的。

  • Trie性质

好多人说trie的根节点不包含任何字符信息,我所习惯的trie根节点却是包含信息的,而且认为这样也方便,下面说一下它的性质 (基于本文所讨论的简单trie树)

1.    字符的种数决定每个节点的出度,即branch数组(空间换时间思想)

2.    branch数组的下标代表字符相对于a的相对位置

3.    采用标记的方法确定是否为字符串。

4.    插入、查找的复杂度均为O(len),len为字符串长度

  • Trie的示意图

如图所示,该trie树存有abc、d、da、dda四个字符串,如果是字符串会在节点的尾部进行标记。没有后续字符的branch分支指向NULL

  • Trie的优点举例

已知n个由小写字母构成的平均长度为10的单词,判断其中是否存在某个串为另一个串的前缀子串。下面对比3种方法:

1.    最容易想到的:即从字符串集中从头往后搜,看每个字符串是否为字符串集中某个字符串的前缀,复杂度为O(n^2)。

2.    使用hash:我们用hash存下所有字符串的所有的前缀子串。建立存有子串hash的复杂度为O(n*len)。查询的复杂度为O(n)* O(1)= O(n)。

3.    使用trie:因为当查询如字符串abc是否为某个字符串的前缀时,显然以b,c,d....等不是以a开头的字符串就不用查找了。所以建立trie的复杂度为O(n*len),而建立+查询在trie中是可以同时执行的,建立的过程也就可以成为查询的过程,hash就不能实现这个功能。所以总的复杂度为O(n*len),实际查询的复杂度只是O(len)。

解释一下hash为什么不能将建立与查询同时执行,例如有串:911,911456输入,如果要同时执行建立与查询,过程就是查询911,没有,然后存入9、91、911,查询911456,没有然后存入9114、91145、911456,而程序没有记忆功能,并不知道911在输入数据中出现过。所以用hash必须先存入所有子串,然后for循环查询。

而trie树便可以,存入911后,已经记录911为出现的字符串,在存入911456的过程中就能发现而输出答案;倒过来亦可以,先存入911456,在存入911时,当指针指向最后一个1时,程序会发现这个1已经存在,说明911必定是某个字符串的前缀。


注:原文末尾还有一段代码和讨论,我就不复制过来了。

抱歉!评论已关闭.