【AHOI 2013】差异

传送门

---

Problem

一个长度为 $n$ 的字符串 $S$，令 $T_i$ 表示它从第 $i$ 个字符开始的后缀。求

$$\sum _{1\le i<j\le n}len(Ti)+len(Tj)-2\times lcp(Ti,Tj)$$

其中，$len(a)$ 表示字符串 $a$ 的长度，$lcp(a,b)$ 表示字符串 $a$ 和字符串 $b$ 的最长公共前缀。

---

Solution

首先有一个结论，即原串建后缀自动机的 $\mathrm{p}\mathrm{a}\mathrm{r}\mathrm{e}\mathrm{n}\mathrm{t}\_\mathrm{t}\mathrm{r}\mathrm{e}\mathrm{e}$ 就是反串的后缀树。

两个后缀的最长公共前缀，转换成两个前缀的最长公共后缀，就是 $\mathrm{p}\mathrm{a}\mathrm{r}\mathrm{e}\mathrm{n}\mathrm{t}\_\mathrm{t}\mathrm{r}\mathrm{e}\mathrm{e}$ 上的 $lca$。

我们将以 $p$ 为儿子的边权值设为 $len(p)-len(f{a}_p)$。如果把 $x,y$ 路径上的边权加起来，答案就是 $len(x)+len(y)-2\times len(lc{a}_{x,y})$，也就是 $len(x)+len(y)-2\times lcp(lc{a}_{x,y})$。

这时问题就变成求树上所有路径的权值和，枚举每条边单独计算贡献即可。

---

Code

#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 1000005
#define ll long long
using namespace std;
char S[N];
int last,tot,Size[N],A[N],sum[N];
ll ans=0;
struct SAM{
	int len,link,nxt[30];
}a[N];
void Insert(int c){
	int p,cur=++tot;
	a[cur].len=a[last].len+1,Size[cur]++;
	for(p=last;p!=-1&&!a[p].nxt[c];p=a[p].link)  a[p].nxt[c]=cur;
	if(p==-1)  a[cur].link=0;
	else{
		int now=a[p].nxt[c];
		if(a[now].len==a[p].len+1)
			a[cur].link=now;
		else{
			int Clone=++tot;
			a[Clone]=a[now],a[Clone].len=a[p].len+1;
			for(;p!=-1&&a[p].nxt[c]==now;p=a[p].link)
				a[p].nxt[c]=Clone;
			a[cur].link=a[now].link=Clone;
		}
	}
	last=cur;
}
int main(){
	scanf("%s",S+1);
	int i,l=strlen(S+1);
	a[0].link=-1;
	for(i=1;i<=l;++i)  Insert(S[i]-'a');
	for(i=1;i<=tot;++i)  sum[a[i].len]++;
	for(i=1;i<=tot;++i)  sum[i]+=sum[i-1];
	for(i=1;i<=tot;++i)  A[sum[a[i].len]--]=i;
	for(i=tot;i>=0;--i){
		Size[a[A[i]].link]+=Size[A[i]];
		ans+=(ll)(a[A[i]].len-a[a[A[i]].link].len)*Size[A[i]]*(l-Size[A[i]]);
	}
	printf("%lld",ans);
	return 0;
}