将原串倒过来,每次添加一个字符相当于增加一个后缀。
问题转化为向集合中动态添加后缀求本质不同的字串的个数,离线求出
SA
找出当前添加的串与集合中的串的最大的
LCP
,就是重复出现的子串的个数,线段树维护集合中rank的前驱和后继,
考场上的原代码(SDOI唯一A掉的一道题QAQ)
#include<algorithm>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<cstdio>
#include<vector>
#include<cmath>
#include<queue>
#include<set>
#include<map>
#define N 100555
using namespace std;
int sc()
{
int i=0,f=1;char c=getchar();
while(c>'9'||c<'0'){if(c=='-')f=-1;c=getchar();}
while(c>='0'&&c<='9')i=i*10+c-'0',c=getchar();
return i*f;
}
struct W{int x,p;}b[N];
int st[N][19],pre[N*6],nxt[N*6];
int a[N],t1[N],t2[N],cc[N],sa[N],rank[N],height[N];
int n,cnt,len;
long long ans[N];
bool cmp(W a,W b)
{
return a.x<b.x;
}
bool cmp1(int *y,int a,int b,int k)
{
int a1=y[a],b1=y[b];
int a2=a+k>=len?-1:y[a+k];
int b2=b+k>=len?-1:y[b+k];
return a1==b1&&a2==b2;
}
void make_sa()
{
int *x=t1,*y=t2,m=len+1;
for(int i=0;i<len;i++)++cc[x[i]=a[i]];
for(int i=1;i<m;i++)cc[i]+=cc[i-1];
for(int i=len-1;i>=0;i--)sa[--cc[x[i]]]=i;
for(int k=1;k<len;k<<=1)
{
int p=0;
for(int i=len-k;i<len;i++)y[p++]=i;
for(int i=0;i<len;i++)
if(sa[i]>=k)y[p++]=sa[i]-k;
for(int i=0;i<m;i++)cc[i]=0;
for(int i=0;i<len;i++)++cc[x[y[i]]];
for(int i=1;i<m;i++)cc[i]+=cc[i-1];
for(int i=len-1;i>=0;i--)sa[--cc[x[y[i]]]]=y[i];
swap(x,y);x[sa[0]]=0;m=1;
for(int i=1;i<len;i++)x[sa[i]]=cmp1(y,sa[i],sa[i-1],k)?m-1:m++;
if(m>=len)break;
}
for(int i=0;i<len;i++)rank[sa[i]]=i;
}
void make_height()
{
int k=0;
for(int i=0;i<len;i++)
{
if(!rank[i])continue;
int j=sa[rank[i]-1];
if(k)k--;
while(a[i+k]==a[j+k])k++;
height[rank[i]]=k;
}
}
void make_st()
{
for(int i=0;i<len;i++)st[i][0]=height[i];
for(int k=1;(1<<k)<len;k++)
for(int i=0;i<len;i++)
if(i+(1<<k)>len)break;
else st[i][k]=min(st[i][k-1],st[i+(1<<k-1)][k-1]);
}
int ask_pre(int x,int l,int r,int p)
{
if(pre[x]==-1||r<=p)return pre[x];
int mid=l+r>>1;
if(p<=mid)return ask_pre(x<<1,l,mid,p);
else
{
int ls=ask_pre(x<<1,l,mid,p);
int rs=ask_pre(x<<1|1,mid+1,r,p);
if(ls==-1||rs==-1)return -ls*rs;
else return max(ls,rs);
}
}
int ask_nxt(int x,int l,int r,int p)
{
if(nxt[x]==-1||p<=l)return nxt[x];
int mid=l+r>>1;
if(p>mid)return ask_nxt(x<<1|1,mid+1,r,p);
else
{
int ls=ask_nxt(x<<1,l,mid,p);
int rs=ask_nxt(x<<1|1,mid+1,r,p);
if(ls==-1||rs==-1)return -ls*rs;
else return min(ls,rs);
}
}
void push_up(int x)
{
int l=x<<1,r=x<<1|1;
if(pre[l]==-1||pre[r]==-1)
pre[x]=-pre[l]*pre[r];
else
pre[x]=max(pre[l],pre[r]);
if(nxt[l]==-1||nxt[r]==-1)
nxt[x]=-nxt[l]*nxt[r];
else
nxt[x]=min(nxt[l],nxt[r]);
}
void change(int x,int l,int r,int p)
{
if(l==r)
{
pre[x]=nxt[x]=p;
return;
}
int mid=l+r>>1;
if(p>mid)change(x<<1|1,mid+1,r,p);
else change(x<<1,l,mid,p);
push_up(x);
}
int ask_st(int l,int r)
{
if(l>r)swap(l,r);l++;
int k=log2(r-l+1);
return min(st[l][k],st[r-(1<<k)+1][k]);
}
int main()
{
//freopen("incantation.in","r",stdin);
//freopen("incantation.out","w",stdout);
n=sc();
for(int i=n-1;i>=0;i--)
b[i].x=sc(),b[i].p=i;
sort(b,b+n,cmp);
for(int i=0;i<n;i++)
if(b[i].x==b[i-1].x)
a[b[i].p]=cnt;
else
a[b[i].p]=++cnt;
len=n;make_sa();make_height();make_st();
memset(pre,-1,sizeof(pre));
memset(nxt,-1,sizeof(nxt));
for(int i=n-1;i>=0;i--)
{
int l=ask_pre(1,0,n-1,rank[i]);
int r=ask_nxt(1,0,n-1,rank[i]);
int mx=0;
if(l!=-1)mx=max(mx,ask_st(l,rank[i]));
if(r!=-1)mx=max(mx,ask_st(rank[i],r));
change(1,0,n-1,rank[i]);
ans[i]=n-i-mx;
}
for(int i=n-1;i>=0;i--)
ans[i]+=ans[i+1],printf("%lld\n",ans[i]);
return 0;
}