来自蒟蒻XXJ的做题记录
%%%陈老师的语文水平
广义后缀自动机的板子题
最多有\(20\)个叶子
随手画一画 以每个叶子为根进行\(dfs\) 到达叶子的路径的后缀 全都是不同的后缀
直接建出来看一下有多少不一样的子串就好了
#include
#define mem(i,j) memset(i,j,sizeof(i))
#define mcy(i,j) memcpy(i,j,sizeof(i))
#define GO(i,here) for(int i=head[here];i!=-1;i=nex[i])
using namespace std;
const int MAXN=100010;
inline int in(){
int a(0);char c=getchar();
while(c<'0'||c>'9') c=getchar();
while(c>='0'&&c<='9') a=(a<<1)+(a<<3)+c-'0',c=getchar();
return a;
}
int n,c,ans;
int head[MAXN],to[2*MAXN],nex[2*MAXN],top1;
int ru[MAXN],col[MAXN];
void add(int a,int b){
nex[top1]=head[a];head[a]=top1;to[top1++]=b;
nex[top1]=head[b];head[b]=top1;to[top1++]=a;
}
struct SAM{
int trans[40*MAXN][11],fa[40*MAXN],lth[40*MAXN];
long long s[40*MAXN];
int cnt;
SAM(){
cnt=0;++cnt;
mem(trans,0);mem(fa,0);mem(lth,0);mem(s,0);
}
int extend(int x,int ww){
int np=++cnt,p=x;
lth[np]=lth[p]+1;
for(;p&&!trans[p][ww];p=fa[p]) trans[p][ww]=np;
if(!p) fa[np]=1;
else{
int q=trans[p][ww];
if(lth[q]==lth[p]+1) fa[np]=q;
else{
int nq=++cnt;
lth[nq]=lth[p]+1;
mcy(trans[nq],trans[q]);
fa[nq]=fa[q];
fa[q]=fa[np]=nq;
for(;p&&trans[p][ww]==q;p=fa[p]) trans[p][ww]=nq;
}
}
return np;
}
void solve(){
for(int i=1;i<=cnt;i++)
ans+=lth[i]-lth[fa[i]];
}
}sam;
void dfs(int here,int f,int x){
int t=sam.extend(x,col[here]);
GO(i,here){
if(to[i]==f) continue;
dfs(to[i],here,t);
}
}
void input(){
mem(head,-1);
n=in();c=in();
for(int i=1;i<=n;i++) col[i]=in();
for(int i=1;isam.lth[j];
}
void init(){
for(int i=1;i<=sam.cnt;i++) rk[i]=i;
sort(rk+1,rk+sam.cnt+1,cmp);
for(int i=1;i<=sam.cnt;i++){
int ind=rk[i];
sam.s[ind]=1;
for(int j=0;j