【介绍】
Trie,又称前缀树或字典树,是一种有序树,用于保存关联数组,其中的键通常是字符串。与二叉查找树不同,键不是直接保存在节点中,而是由节点在树中的位置决定。
【思想】
举个例子比如说is,him,his,her,he
其中的红色节点就是所对应的字符串的末尾,也就是累计权值的地方。
定义数组
son[i][j]
表示节点编号为
i
,字符编号为
所以Trie的空间复杂度为O(字符串的总长*字符的种类数)Ps:可以用邻接表降低空间复杂度,但会升高时间复杂度。
【核心代码】
struct Tire
{
int son[maxv][maxt],w[maxv],len;
int Getid(char ch) {return ch-'a';}//这里博主写的是字符种类只有小写字母的。
void Insert(char *s)//加入一个字符串
{
int now=0;
for (int i=1; s[i]; i++)
{
int p=Getid(s[i]);
if (!son[now][p]) son[now][p]=++len;
now=son[now][p];
}
w[now]++;
}
int Getpre(char *s)//统计前缀总数
{
int now=0,sum=0;
for (int i=1; s[i]; i++)
{
int p=Getid(s[i]); sum+=w[now];
if (son[now][p]) now=son[now][p]; else break;
}
return sum;
}
}tr;