bzoj3238: [Ahoi2013]差异

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<stack>
using namespace std;
#define REP(i,s,t) for(int i=s;i<=t;i++)
#define dwn(i,s,t) for(int i=s;i>=t;i--)
#define clr(x,c) memset(x,c,sizeof(x))
#define ll long long
const int nmax=500005;
const int inf=0x7f7f7f7f;

char S[nmax];int c[nmax],t[nmax],t2[nmax],rank[nmax],sa[nmax],height[nmax],n,m;
void build_sa(){
	int m=27,*x=t,*y=t2;
	for(int i=0;i<m;i++) c[i]=0;
	for(int i=0;i<n;i++) c[x[i]=S[i]]++;
	for(int i=1;i<m;i++) c[i]+=c[i-1];
//...	for(int i=0;i<n;i++) sa[--c[x[i]]]=i;
    for(int i=n-1;i>=0;i--) sa[--c[x[i]]]=i;
	for(int k=1;k<=n;k<<=1){
		int p=0;
		for(int i=n-k;i<n;i++) y[p++]=i;
		for(int i=0;i<n;i++) if(sa[i]>=k) y[p++]=sa[i]-k;
		for(int i=0;i<m;i++) c[i]=0;
		for(int i=0;i<n;i++) c[x[i]]++;
		for(int i=1;i<m;i++) c[i]+=c[i-1];
		for(int i=n-1;i>=0;i--) sa[--c[x[y[i]]]]=y[i];
		swap(x,y);p=1;x[sa[0]]=0;
		for(int i=1;i<n;i++) x[sa[i]]=y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k]?p-1:p++;
		if(p>=n) break;
		m=p;
	}
}

void build_height(){
	int k=0;
	for(int i=0;i<n;i++) rank[sa[i]]=i;
	for(int i=0;i<n;i++){
		if(k) k--;
		int j=sa[rank[i]-1];
		while(S[i+k]==S[j+k]) k++;
		height[rank[i]]=k;
	}
}

int l[nmax],r[nmax];
void work(){
	stack<int>s;
	l[1]=1;s.push(1);
	for(int i=2;i<n;i++) {
		while(!s.empty()&&height[s.top()]>height[i]) s.pop();
		if(s.empty()) l[i]=1;else l[i]=s.top()+1;
		s.push(i);
	}
	while(!s.empty()) s.pop();
	r[n-1]=n-1;s.push(n-1);
	for(int i=n-2;i;i--){
		while(!s.empty()&&height[s.top()]>=height[i]) s.pop();
		if(s.empty()) r[i]=n-1;else r[i]=s.top()-1;
		s.push(i);
	}
	ll ans=(ll)n*(n-1)*(n-2)/2;
	for(int i=1;i<n;i++) ans-=(ll)height[i]*(i-l[i]+1)*(r[i]-i+1)*2;
	printf("%lld\n",ans);
}

int main(){
	scanf("%s",S);n=strlen(S);
	for(int i=0;i<n;i++) S[i]=S[i]-‘a‘+1;
	S[n++]=0;
	build_sa();build_height();
	work();return 0;
}