最开始想到的是SA RMQ预处理,不过既然在写SAM那就用SAM来解决一下这个问题
其实写起来SAM远比SA简洁
我们这样想,对于两个后缀,他的lcp的长度就是两个对应的接收态在fail树上的LCA的深度
如果已经想明白了上面这点,那么对于这个问题,算得就是对于所有节点,求他的所有子节点对的LCA的sigma
在下面的代码中,除了模板部分就添加了两个数组,rig代表right集合,pre代表rig的遍历方向的前缀和,树上随便dp一下就能处理出pre了,ans也就很明显了。
#pragma GCC optimize(2)
#pragma GCC optimize(3)
#pragma GCC optimize(4)
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 5e5 + 10;
struct SAM
{
static const int KN = N << 1;
static const int KM = 30;
int fail[KN], net[KN][KM], len[KN], cnt, root;
int pre[KN],rig[KN];
int newnode(int _len)
{
memset(net[cnt], -1, sizeof(net[cnt]));
//fail[cnt] = -1;
len[cnt] = _len;
return cnt++;
}
void init()
{
cnt = 0;
memset(fail,-1,sizeof(fail));
root = newnode(0);
}
int add(int p, int x)
{
// int np = newnode(len[p] + 1);
int np=newnode(len[p]+1);
pre[np]=1,rig[np]=1;
while(~p && net[p][x] == -1) net[p][x] = np, p = fail[p];
if(p == -1) fail[np] = root;
else
{
int q = net[p][x];
if(len[q] == len[p] + 1) fail[np] = q;
else
{
int nq = newnode(len[p] + 1);
memcpy(net[nq], net[q], sizeof(net[q]));
fail[nq] = fail[q];
fail[q] = fail[np] = nq;
while(~p && net[p][x] == q) net[p][x] = nq, p = fail[p];
}
}
// ccnt[np][sy]++;
return np;
}
void build(char *s, char ch)
{
int now = root;
for(int i = 0; s[i]; ++i) now = add(now, s[i] - ch);
}
int ord[KN], pri[KN];
void topo()
{
int maxVal=0;
memset(pri, 0, sizeof(pri));
for (int i = 0; i < cnt; ++i) maxVal = max(maxVal, len[i]), ++ pri[len[i]];
for (int i = 1; i <= maxVal; ++i) pri[i] += pri[i - 1];
for (int i = 0; i < cnt; ++i) ord[--pri[len[i]]] = i;
}
void gao(ll ans)
{
topo();
for(int i=cnt-1;i>=0;i--) rig[fail[ord[i]]]+=rig[ord[i]];
for(int i=0;i<cnt;i++) ans-=2ll*pre[fail[i]]*rig[i]*len[fail[i]],pre[fail[i]]+=rig[i];
printf("%lld\n",ans);
}
} sam;
char s[N];
int main()
{
scanf("%s",s);
int le=strlen(s);
ll ans=1ll*(1+le)*le/2*(le-1);
// cout<<ans<<endl;
sam.init();
sam.build(s,'a');
sam.gao(ans);
}