BZOJ3926

来自蒟蒻XXJ的做题记录

%%%陈老师的语文水平

广义后缀自动机的板子题

最多有\(20\)个叶子

随手画一画 以每个叶子为根进行\(dfs\) 到达叶子的路径的后缀 全都是不同的后缀

直接建出来看一下有多少不一样的子串就好了

#include<bits/stdc++.h>
#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;i<n;i++){
		int a,b;
		a=in();b=in();
		add(a,b);
		++ru[a];++ru[b];
	}
	for(int i=1;i<=n;i++)
		if(ru[i]==1) dfs(i,0,1);
}
int rk[40*MAXN];
bool cmp(int i,int j){
	return sam.lth[i]>sam.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<c;j++){
			if(sam.trans[ind][j]) sam.s[ind]+=sam.s[sam.trans[ind][j]];
		}
	}
}
void xxj(){
	//sam.solve();
	init();
}
void output(){
	cout<<sam.s[1]-1<<endl;
}

int main(){
	input();
	xxj();
	output();
	return 0;
}