在后面添加字符的操作很不舒服,可以倒过来看成是添加后缀。每加入一个后缀,在统计的时候需要去掉之前已经出现过的前缀。在SA上面找前后最近的已经被加入的位置求LCP即可。找最近的位置可以用set维护。
#include<cstdio>
#include<cstring>
#include<set>
#include<algorithm>
using namespace std;
#define LL long long
const int maxn=200010;
set<int> s;
int sa[maxn],rank[maxn],f[maxn],cnt[maxn],a[maxn],ord[maxn],height[maxn],
mn[maxn][20],log[maxn],n,m;
int qry(int l,int r)
{
int k=log[r-l+1];
return min(mn[l][k],mn[r-(1<<k)+1][k]);
}
int main()
{
//freopen("in.txt","r",stdin);
//freopen("out.txt","w",stdout);
int p,now,x;
LL ans=0;
scanf("%d",&n);
for (int i=1;i<=n;i++)
{
scanf("%d",&a[n-i+1]);
ord[i]=a[n-i+1];
}
sort(ord+1,ord+n+1);
m=unique(ord+1,ord+n+1)-ord-1;
for (int i=1;i<=n;i++)
f[i]=lower_bound(ord+1,ord+m+1,a[i])-ord;
for (int i=1;i<=n;i++) cnt[rank[i]=f[i]]++;
for (int i=1;i<=m;i++) cnt[i]+=cnt[i-1];
for (int i=n;i;i--) sa[cnt[f[i]]--]=i;
for (int k=1;;k++)
{
p=0;
for (int i=n-(1<<k-1)+1;i<=n;i++) f[++p]=i;
for (int i=1;i<=n;i++)
if (sa[i]-(1<<k-1)>0)
f[++p]=sa[i]-(1<<k-1);
for (int i=1;i<=m;i++) cnt[i]=0;
for (int i=1;i<=n;i++) cnt[rank[f[i]]]++;
for (int i=1;i<=m;i++) cnt[i]+=cnt[i-1];
for (int i=n;i;i--) sa[cnt[rank[f[i]]]--]=f[i];
for (int i=1;i<=n;i++) f[i]=rank[i];
rank[sa[1]]=1;
for (int i=2;i<=n;i++)
if (f[sa[i]]!=f[sa[i-1]]||f[sa[i]+(1<<k-1)]!=f[sa[i-1]+(1<<k-1)])
rank[sa[i]]=rank[sa[i-1]]+1;
else
rank[sa[i]]=rank[sa[i-1]];
m=rank[sa[n]];
if (m>=n) break;
}
for (int i=1;i<=n;i++)
{
height[rank[i]]=height[rank[i-1]];
if (height[rank[i]]) height[rank[i]]--;
while (a[i+height[rank[i]]]==a[sa[rank[i]-1]+height[rank[i]]])
height[rank[i]]++;
}
for (int k=1;(1<<k)<=n;k++) log[1<<k]=k;
for (int i=3;i<=n;i++)
if (!log[i]) log[i]=log[i-1];
for (int i=1;i<=n;i++) mn[i][0]=height[i];
for (int k=1;k<=log[n];k++)
for (int i=1;i+(1<<k)-1<=n;i++)
mn[i][k]=min(mn[i][k-1],mn[i+(1<<k-1)][k-1]);
s.insert(0);
s.insert(n+1);
for (int i=n;i;i--)
{
now=0;
x=*(--s.lower_bound(rank[i]));
if (x) now=max(now,qry(x+1,rank[i]));
x=*(s.lower_bound(rank[i]));
if (x<=n) now=max(now,qry(rank[i]+1,x));
ans+=n-i+1-now;
s.insert(rank[i]);
printf("%lld\n",ans);
}
}