【AHOI2013】差异【后缀自动机】

后缀自动机总结

传送门

求任意两个后缀长度减去其公共前缀长度。

如果把串反过来，就是任意两个前缀长度减去其公共后缀长度。

公共后缀直接fail树上lca即可。

而字符串前缀只有原串长度len个。

根据公式，如果将问题转化为两点路径长度和，那边权赋为子节点长减去fail节点长。这样问题就得以转化。

而一条边的贡献就是其两端size的乘积。

如此可以计算。

#include<bits/stdc++.h>
using namespace std;
#define in read()
#define int long long
int in{
	int cnt=0,f=1;char ch=0;
	while(!isdigit(ch)){
		ch=getchar();if(ch=='-')f=-1;
	}
	while(isdigit(ch)){
		cnt=cnt*10+ch-48;
		ch=getchar();
	}return cnt*f;
}
struct node{
	int ch[26],len,fail;
}t[1000003];int cnt=1,last=1;
int first[1000003],nxt[1000003],to[1000003],w[1000003],tot;
void add(int a,int b,int c){
	nxt[++tot]=first[a];first[a]=tot;to[tot]=b;w[tot]=c;
}int size[1000003];
char ch[500003];int len;int now;
void insert(int x){
	int p=last;last=now=++cnt;t[now].len=t[p].len+1;size[now]++;
	for(;p&&!t[p].ch[x];p=t[p].fail)t[p].ch[x]=now;
	if(!p)t[now].fail=1;
	else{
		int q=t[p].ch[x];if(t[q].len==t[p].len+1)t[now].fail=q;
		else{
			int temp=++cnt;t[temp]=t[q];t[temp].len=t[p].len+1;
			t[now].fail=t[q].fail=temp;for(;p&&t[p].ch[x]==q;p=t[p].fail)t[p].ch[x]=temp;
		}
	}
}int ans;
void dfs(int u,int fa){
	for(int i=first[u];i;i=nxt[i]){
		int v=to[i];dfs(v,u);
		//cout<<w[i]<<" "<<size[v]<<" "<<cnt-size[v]<<endl;
		ans+=w[i]*size[v]*(len-size[v]);
		size[u]+=size[v];
	}
}
signed main(){
	scanf("%s",ch+1);len=strlen(ch+1);
	for(int i=len;i>=1;i--)insert(ch[i]-'a');
	for(int i=2;i<=cnt;i++)add(t[i].fail,i,t[i].len-t[t[i].fail].len);
	dfs(1,0);cout<<ans;
	return 0;
}