luogu3346 诸神眷顾的幻想乡 (广义SAM)

首先，让每一个叶节点做一次树根的话，每个路径一定至少有一次会变成直上直下的

于是对于每个叶节点作为根产生的20个trie树，把它们建到同一个广义SAM里

建法是对每个trie dfs去建，last就是父亲的那个节点；每次做一个新trie时，last给成root

然后答案就是每个节点表示的长度和

#include<bits/stdc++.h>
#define pa pair<int,int>
#define CLR(a,x) memset(a,x,sizeof(a))
#define MP make_pair
using namespace std;
typedef long long ll;
const int maxn=1e5+10,maxp=4e6+10;

inline char gc(){
    return getchar();
    static const int maxs=1<<16;static char buf[maxs],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,maxs,stdin),p1==p2)?EOF:*p1++;
}
inline ll rd(){
    ll x=0;char c=gc();bool neg=0;
    while(c<'0'||c>'9'){if(c=='-') neg=1;c=gc();}
    while(c>='0'&&c<='9') x=(x<<1)+(x<<3)+c-'0',c=gc();
    return neg?(~x+1):x;
}

int N,C,eg[maxn*2][2],egh[maxn],ect,dgr[maxn];
int col[maxn];
int tr[maxp][12],fa[maxp],len[maxp],pct=1,rt=1;

inline void adeg(int a,int b){
    eg[++ect][0]=b,eg[ect][1]=egh[a],egh[a]=ect;
    dgr[a]++;
}

inline int insert(int x,int o){
    int p=++pct;
    len[p]=len[o]+1;
    for(;o&&!tr[o][x];o=fa[o]) tr[o][x]=p;
    if(!o){fa[p]=rt;return p;}
    int q=tr[o][x];
    if(len[q]==len[o]+1){fa[p]=q;return p;}
    int qq=++pct;
    memcpy(tr[qq],tr[q],sizeof(tr[qq]));
    len[qq]=len[o]+1;fa[qq]=fa[q],fa[q]=fa[p]=qq;
    for(;o&&tr[o][x]==q;o=fa[o]) tr[o][x]=qq;
    return p;
}

inline void dfs(int x,int f,int p){
    p=insert(col[x],p);
    for(int i=egh[x];i;i=eg[i][1]){
        int b=eg[i][0];if(b==f) continue;
        dfs(b,x,p);    
    }
}

int main(){
    //freopen("","r",stdin);
    int i,j,k;
    N=rd(),C=rd();
    for(i=1;i<=N;i++) col[i]=rd();
    for(i=1;i<N;i++){
        int a=rd(),b=rd();
        adeg(a,b);adeg(b,a);
    }
      for(i=1;i<=N;i++){
          if(dgr[i]==1) dfs(i,0,rt);
      }
      ll ans=0;
      for(i=2;i<=pct;i++) ans+=len[i]-len[fa[i]];
    printf("%lld
",ans);
    return 0;
}