【bzoj3926】[Zjoi2015]諸神眷顧的幻想鄉
阿新 • • 發佈:2019-02-03
哼要先給那些根本沒寫清楚還發表的題解差評0.0如果不是clj的題解根本看不懂好嘛0.0
題意求一棵樹上本質不同的子串的數量 每個點出度<=20 葉子節點<=20。
從所有的葉子節點開始到根的每條路徑的每一個字首都對應一個子串(公共路徑上會重複)
字尾自動機上可以識別全部的字尾 也就是說按照從葉子節點到根的順序插進去就相當於統計了所有的字首 於是就想到了字尾自動機~
字尾自動機不同的狀態一定對應不同的子串並且整個自動機對應的一定是全部的子串 所以ans+=len[i]-len[fa[i]]就是答案.
其實這個題用到的就是廣義自動機 就是對一棵tire樹建立一個自動機
跟一般的字尾自動機的區別就是 其實沒什麼區別 就相當於從每一個葉子節點到根的路徑上的點都從parent樹根開始插入一遍 這樣重複的部分一定會在自動機中合併
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<map>
using namespace std;
const int N=400011;
int head[N],next[N],key[N],tot;
int a[N],du[N];
int last=1,cnt=1;
long long ans=0;
struct node
{
int a[N*10][15],fa[N*10],len[N*10];
int insert(int p,int x)
{
int np;
np=last=++cnt;
len[np]=len[p]+1;
while(!a[p][x]&&p)a[p][x]=np,p=fa[p];
if(!p)fa[np]=1;
else
{
int q=a[p][x];
if(len[p]+1==len[q])fa[np]=q;
else
{
int nq=++cnt;len[nq]=len[p]+1;
memcpy(a[nq],a[q],sizeof (a[q]));
fa[nq]=fa[q];
fa[np]=fa[q]=nq;
while(a[p][x]==q)a[p][x]=nq,p=fa[p];
}
}
return np;
}
void get()
{
for(int i=1;i<=cnt;i++)
ans+=len[i]-len[fa[i]];
}
}sam;
void add(int x,int y)
{
tot++;
next[tot]=head[x];
head[x]=tot;
key[tot]=y;
}
void dfs(int x,int fa,int p)
{
int t=sam.insert(p,a[x]);
for(int i=head[x];i;i=next[i])
{
int y=key[i];
if(y!=fa)dfs(y,x,t);
}
}
int main()
{
int n,c;cin>>n>>c;
for(int i=1;i<=n;i++)scanf("%d",&a[i]);
for(int i=1;i<n;i++)
{
int x,y;
scanf("%d%d",&x,&y);
add(x,y);add(y,x);
du[x]++,du[y]++;
}
for(int i=1;i<=n;i++)
if(du[i]==1)dfs(i,0,1);
sam.get();
printf("%lld\n",ans);
}