题目描述
给定一个长度为
n
n
的字符串,令
Ti
T
i
表示它从第
i
i
个字符开始的后缀。求
其中, len(a) l e n ( a ) 表示字符串 a a 的长度,表示字符串 a a 和字符串的最长公共前缀。
输入输出格式
输入格式:
一行,一个字符串
S
S
。
输出格式:
一行,一个整数,表示所求值。
输入输出样例
输入样例#1:
cacao
输出样例#1:
54
说明
对于 100% 的数据,保证 ,且均为小写字母。
分析:
主要就是求每两个子串的
lcp
l
c
p
。
两个串
lcp
l
c
p
的长度就是
fail
f
a
i
l
树上的
lca
l
c
a
。
这个东西还没搞懂,所以就不解释了。
代码:
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#define LL long long
const int maxn=1e6+7;
using namespace std;
int n,cnt,size[maxn],b[maxn],top[maxn],sum[maxn];
char s[maxn];
LL ans;
struct node{
int len,fail;
int son[26];
}t[maxn];
void build_sam()
{
cnt=1;
int now=1,p,q,clone;
for (int i=0;i<n;i++)
{
int c=s[i]-'a';
p=now;
now=++cnt;
t[now].len=t[p].len+1;
size[now]=sum[now]=1;
while (p&&(!t[p].son[c])) t[p].son[c]=now,p=t[p].fail;
if (!p) t[now].fail=1;
else
{
q=t[p].son[c];
if (t[p].len+1==t[q].len) t[now].fail=q;
else
{
clone=++cnt;
t[clone]=t[q];
t[clone].len=t[p].len+1;
t[q].fail=t[now].fail=clone;
while (p&&(t[p].son[c]==q)) t[p].son[c]=clone,p=t[p].fail;
}
}
}
}
void calc()
{
for (int i=1;i<=cnt;i++) b[t[i].len]++;
for (int i=1;i<=n;i++) b[i]+=b[i-1];
for (int i=1;i<=cnt;i++) top[b[t[i].len]--]=i;
for (int i=cnt;i>0;i--)
size[t[top[i]].fail]+=size[top[i]];
ans=0;
for (int i=1;i<=n;i++) ans+=(LL)i*(n-1);
for (int i=cnt;i>0;i--)
{
ans-=(LL)sum[t[i].fail]*size[i]*t[t[i].fail].len*2;
sum[t[i].fail]+=size[i];
}
printf("%lld",ans);
}
int main()
{
scanf("%s",s);
n=strlen(s);
build_sam();
calc();
}