codeforces 802I 后缀自动机

题意：给你一个字符串，统计他的所有不同子串的出现次数的平方和，n<=100000

其实相当于一个SAM的模板题吧。

首先你要理解SAM的存储方式以后就可以轻松切这题。。

SAM建立了一个parent树还有一个DAG，这里显然和DAG没什么关系。

对于parent树上的每个节点，他有一个right集，存储这一些子串的右端点下标，同时还有一个对应区间min(len),max(len)

那么对于一个right集合中的每一个长度len，他和所有右端点配对以后形成的子串是一样的，这意味着right集合的大小就是这个长度为len的子串的出现次数。。

那么这个right集合的大小就是r−l，所以每个集合的贡献就是|right|2∗(r−l)

right集的大小很容易算，至于l,根据parent树的性质，就是他父亲的右端点。

不知道为什么代码不高亮了，辣鸡csdn。。建议贴回自己编译器里面。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
#define N 500005
#define ll long long
using namespace std;
char s
;
int S,cnt,last;
int a[N],b[N],f[N],t[N],fa[N],l[N],r[N],ch[N
18658
][26];
void add(int x)
{
int c=a[x];
int p=last,np=++cnt;last=np;
l[np]=x;
for(;p&&!ch[p][c];p=fa[p])ch[p][c]=np;
if(!p)fa[np]=S;
else
{
int q=ch[p][c];
if(l[p]+1==l[q])fa[np]=q;
else
{
int nq=++cnt;l[nq]=l[p]+1;
memcpy(ch[nq],ch[q],sizeof ch[q]);
fa[nq]=fa[q];
fa[np]=fa[q]=nq;
for(;ch[p][c]==q;p=fa[p])ch[p][c]=nq;
}
}
}
int main()
{
int cas;
scanf("%d",&cas);
while (cas--)
{
cnt=0;
memset(r,0,sizeof(r));
memset(l,0,sizeof(l));
memset(ch,0,sizeof(ch));
memset(fa,0,sizeof(fa));
memset(t,0,sizeof(t));
scanf("%s",s+1);

last=S=++cnt;
int len=strlen(s+1);
for(int i=1;i<=len;i++)a[i]=s[i]-'a';
for(int i=1;i<=len;i++)add(i);
for(int i=1,p=S;i<=len;i++)
{
p=ch[p][a[i]];r[p]++;
}
for(int i=1;i<=cnt;i++)b[i]=0;
for(int i=1;i<=cnt;i++)b[l[i]]++;
for(int i=1;i<=len;i++)b[i]+=b[i-1];
for(int i=1;i<=cnt;i++)t[b[l[i]]--]=i;
for(int i=cnt;i;i--)r[fa[t[i]]]+=r[t[i]];
//for(int i=1;i<=cnt;i++)f[l[i]]=max(f[l[i]],r[i]);
//for(int i=len;i;i--)f[i]=max(f[i+1],f[i]);
long long ans=0;
for(int i=1;i<=cnt;i++)ans+=(long long)r[i]*r[i]*(l[i]-l[fa[i]]);
printf("%lld\n",ans);
}
return 0;
}