题目:
题解:
广义的后缀自动机就是对多串建立的后缀自动机。
对于这道题目来说,要求不同的子串个数,这个不就是裸的step[i]-step[fa[i]]吗,但是由于还有”顺序“问题(红黄蓝和蓝黄红),那么我们就需要对于不同的顺序建立后缀自动机,我们发现题目说叶节点不超过20个,那么我们可以从叶节点开始每次遍历树建立后缀自动机,这样得到了所有的链(即获得了所有可能的顺序),然后在后缀自动机上奔跑就可以了
trie 树的后缀自动机的状态数是线性的,不超过2|T|,|T|应该是trie的节点数。
trie树的后缀自动机的转移函数(边数)的上界为trie的节点数*字符集大小。
注意这东西既然是一棵trie树而不是很多不一样的字符串,我们就不能把last每次清1带进去玩了,而是按照树的结构一样添加东西
代码:
#include <cstdio>
#include <cstring>
#define LL long long
using namespace std;
const int N=2000005;
int np,p,nq,q,cnt,ch[N*2][15],fa[N],tot,nxt[N],point[N],v[N],c[N],step[N],du[N];
LL ans;
void addline(int x,int y)
{
++tot; nxt[tot]=point[x]; point[x]=tot; v[tot]=y;
++tot; nxt[tot]=point[y]; point[y]=tot; v[tot]=x;
}
int insert(int p,int c)
{
np=++cnt; step[np]=step[p]+1;
while (p && !ch[p][c]) ch[p][c]=np,p=fa[p];
if (!p) {fa[np]=1;return np;}
q=ch[p][c];
if (step[q]==step[p]+1){fa[np]=q;return np;}
nq=++cnt; step[nq]=step[p]+1;
memcpy(ch[nq],ch[q],sizeof(ch[q]));
fa[nq]=fa[q]; fa[q]=fa[np]=nq;
while (ch[p][c]==q) ch[p][c]=nq,p=fa[p];
return np;
}
void dfs(int x,int fa,int p)
{
int t=insert(p,c[x]);
for (int i=point[x];i;i=nxt[i])
if (v[i]!=fa) dfs(v[i],x,t);
}
int main()
{
cnt=1;int n,cc;
scanf("%d%d",&n,&cc);
for (int i=1;i<=n;i++) scanf("%d",&c[i]);
for (int i=1;i<n;i++)
{
int x,y;scanf("%d%d",&x,&y);
addline(x,y); du[x]++; du[y]++;
}
for (int i=1;i<=n;i++)
if (du[i]==1) dfs(i,0,1);
for (int i=1;i<=cnt;i++)
ans+=(LL)(step[i]-step[fa[i]]);
printf("%lld",ans);
}