Trie

Trie树是一个基础数据结构,用来高效的存储和查找字符串,一般Trie是由大写字母,小写字母,0/1构成的,yxc大佬说过,只要是用Trie树做的题目,都不会有太多的种类。

给个例图(借用yxc大佬画的(不想动了QAQ))

标记星号的是证明以那个字母为结尾是有单词的,先放上一道模板题

Trie字符串统计

题面描述

维护一个字符串集合,支持两种操作:

  1. “I x”向集合中插入一个字符串x;
  2. “Q x”询问一个字符串在集合中出现了多少次。

共有N个操作,输入的字符串总长度不超过 105105,字符串仅包含小写英文字母。

输入格式

第一行包含整数N,表示操作数。

接下来N行,每行包含一个操作指令,指令为”I x”或”Q x”中的一种。

输出格式

对于每个询问指令”Q x”,都要输出一个整数作为结果,表示x在集合中出现的次数。

每个结果占一行。

样例

这个题就是Trie树的模板题,查找与插入,一看数据范围就不能进行暴力,只能来用Trie树来进行高效的字符串查找,先建一个根节点(root),然后在插入操作的时候如果有这个字母,就接着往下走,如果没有,就新建一个字母与它相连,依次类推,可以得到上面那幅图,在单词的结尾进行标记。查找时找到结尾如果没有标记输出0,否则输出标记的次数就好。Trie树的基本就是这样,下面上代码

代码出自yxc大佬的模板

void in(char str[])
{
	int p=0;
	for(int i=0;str[i];i++)
	{
		int u=str[i]-'a';
		if(!son[p][u])
		son[p][u]=++idx;
		p=son[p][u];
	}
	cnt[p]++;
}

以上为插入代码

void q(char str[])
{
	int p=0;
	for(int i=0;str[i];i++)
	{
		int u=str[i]-'a';
		if(!son[p][u])
		return 0;
		p=son[p][u];
	}
	return cnt[p];
}

以上为查询代码

插入与查询要理解一下,这道题也就解决了;要自己画图体会;

留下评论

通过 WordPress.com 设计一个这样的站点
从这里开始