【bzoj3238】差异 后缀自动机

AC通道：http://www.lydsy.com/JudgeOnline/problem.php?id=3238

【题解】

题上所给公式可以化为

其中：

=1×(n-1)+2×(n-2)+3×(n-3)+……+(n-1)×1+2×1+3×2+……+n×(n-1)

=(n+1)+2×(n+1)+3×(n+1)+……+(n-1)(n+1)

=(1+2+3+……+(n-1))(n+1)

=n(n-1)(n+1)/2

接着考虑后面的，我们把串逆序后建成SAM，搞出后缀树，我们知道在后缀树上两个点的LCA的深度就是它们的公共字串长度

所以我们考虑树形dp，对于每一条边，它的贡献就是v[i]*C[size[x]][2]

后来膜拜了一下LZX的代码，发现我们其实不用建出后缀树，只需要搞出它的拓扑序，用子节点更新父节点的答案就行了。

比建后缀树的代码快了1s，在bzoj上排名第10，（那些用指针写的真心快啊）

#include
#include
#include
#include
#include
#include
#include
using namespace std;
typedef long long ll;
#define FILE "read"
#define MAXN 1000010
#define up(i,j,n) for(int i=j;i<=n;++i)
#define dn(i,j,n) for(int i=j;i>=n;--i) 
#define cmax(a,b) a=max(a,b)
#define cmin(a,b) a=min(a,b)
//struct node{int y,next,v;}e[MAXN*2];
int n,now(1),cnt(1),len,Link[MAXN],size[MAXN],par[MAXN],c[MAXN],id[MAXN],mx[MAXN],son[MAXN][27];
ll ans;
char ch[MAXN];
/*void insert(int x,int y,int v){
	e[++len].next=Link[x];Link[x]=len;e[len].y=y;e[len].v=v;
	e[++len].next=Link[y];Link[y]=len;e[len].y=x;e[len].v=v;
}*/
void extend(int x){
	int p=now,np=++cnt;
	mx[np]=mx[now]+1; now=np;  size[np]=1;
	while(!son[p][x]&&p)  son[p][x]=np,p=par[p];
	if(!p)  par[np]=1;
	else{
		int q=son[p][x];
		if(mx[q]==mx[p]+1)  par[np]=q;
		else{
			int nq=++cnt; mx[nq]=mx[p]+1;
			memcpy(son[nq],son[q],sizeof(son[q]));
			par[nq]=par[q]; par[q]=par[np]=nq;
			while(son[p][x]==q)  son[p][x]=nq,p=par[p];
		}
	}
}
void topsort(){
	up(i,1,cnt)  c[mx[i]]++;
	up(i,1,n)    c[i]+=c[i-1];
	up(i,1,cnt)  id[c[mx[i]]--]=i;
}
/*void dp(int x,int fa,int v){
	for(int i=Link[x];i;i=e[i].next){
		if(e[i].y==fa)  continue;
		dp(e[i].y,x,e[i].v);
		size[x]+=size[e[i].y];
	}
	ans-=(ll)size[x]*(size[x]-1)*v;
}*/
int main(){
	freopen(FILE".in","r",stdin);
	freopen(FILE".out","w",stdout);
	scanf("%s",ch+1); n=strlen(ch+1);
	dn(i,n,1)  extend(ch[i]-'a');
	//up(i,2,cnt)  insert(i,par[i],mx[i]-mx[par[i]]);
	ans=(ll)n*(n-1)*(n+1)/2;  //dp(1,0,0);
	topsort();  
	dn(i,cnt,2) {
		int x=id[i];
		size[par[x]]+=size[x];
		ans-=(ll)(mx[x]-mx[par[x]])*(size[x]-1)*size[x];
	}
	printf("%lld\n",ans);
	return 0;
}