题目链接
思路:先不看lcp部分,若只有len[i] 则
ans=(n−1)×∑ni=1
然后就变成了一个跑后缀数组得到height值+RMQ的题目
等等……枚举的时间复杂度为O(n^2)明显超时,那么我们需要在O(n)或者O(nlog)的时间内处理出区间的最小值,注意我们并不需要知道具体的每一个区间是多少,只用知道总值就可以了~~
我们用单调栈来维护这个问题,对于任意一个i答案为向左向右拓展长度的乘积
#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<stack>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fod(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
const int N=5e5+10;
typedef long long ll;
int n,p,q,k;
char ch[N];
int a[N],v[N],h[N];
int sa[2][N],rk[2][N]; ll dp[N];
void calsa(int sa[N],int rk[N],int SA[N],int RK[N])
{
for(int i=1;i<=n;i++)v[rk[sa[i]]]=i;
for(int i=n;i;i--)
if(sa[i]>k)
SA[v[rk[sa[i]-k]]--]=sa[i]-k;
for(int i=n-k+1;i<=n;i++)SA[v[rk[i]]--]=i;
for(int i=1;i<=n;i++)
RK[SA[i]]=RK[SA[i-1]]+(rk[SA[i-1]]!=rk[SA[i]]||rk[SA[i]+k]!=rk[SA[i-1]+k]);
}
void getsa()
{
memset(v,0,sizeof(v));
p=0,q=1;
for(int i=1;i<=n;i++)v[a[i]]++;
for(int i=1;i<=30;i++)v[i]+=v[i-1];
for(int i=1;i<=n;i++)
sa[p][v[a[i]]--]=i;
for(int i=1;i<=n;i++)
rk[p][sa[p][i]]=rk[p][sa[p][i-1]]+(a[sa[p][i-1]]!=a[sa[p][i]]);
for(k=1;k<n;k<<=1,swap(p,q))
calsa(sa[p],rk[p],sa[q],rk[q]);
}
void geth()
{
k=0;
for(int i=1;i<=n;i++)
if(rk[p][i]==1)h[rk[p][i]]=0;
else
{
int j=sa[p][rk[p][i]-1];
while(a[i+k]==a[j+k])k++;
h[rk[p][i]]=k;if(k>0)k--;
}
}
stack<int>s;
int main()
{
freopen("b.in","r",stdin);
freopen("b.out","w",stdout);
scanf("%s",ch+1);
n=strlen(ch+1);
fo(i,1,n) a[i]=ch[i]-'a'+1;
getsa(); geth();
ll ans=ll(n+1)*(n-1)*ll(n)>>1;
for(int i=n;i;i--) {
while(!s.empty()&&h[s.top()]>h[i])s.pop();
int tmp; if(!s.empty()) tmp=s.top(); else tmp=n+1;
dp[i]=(ll)h[i]*(tmp-i)+dp[tmp];
ans-=dp[i]*2;
s.push(i);
}
cout<<ans<<endl;
return 0;
}