【BZOJ 3230】相似子串 后缀数组

出题人在下面写题解系列

本质不同的字串个数=∑n-h[i]-sa[i]

然后一个一个套就好了

#include<cstdio>#include<cstring>#include<iostream>#define maxn 200021#define LL long longusing namespace std;struct SA{int n,sa[maxn],rk[maxn],f[maxn*2][18],t1[maxn],t2[maxn];int h[maxn],lg[maxn],cnt[maxn];LL sum[maxn];char s[maxn];void build(int m){int p,i,*x=t1,*y=t2;for(i=0;i<n;i++)cnt[x[i]=s[i]-'a']++;for(i=1;i<m;i++)cnt[i]+=cnt[i-1];for(i=n-1;i>=0;i--)sa[--cnt[x[i]]]=i;for(int k=1;k<n;k<<=1){p=0;for(i=n-k;i<n;i++)y[p++]=i;for(i=0;i<n;i++)if(sa[i]>=k)y[p++]=sa[i]-k;for(i=0;i<=m;i++)cnt[i]=0;for(i=0;i<n;i++)cnt[x[y[i]]]++;for(i=1;i<=m;i++)cnt[i]+=cnt[i-1];for(i=n-1;i>=0;i--)sa[--cnt[x[y[i]]]]=y[i];swap(x,y);p=1,x[sa[0]]=0;for(i=1;i<n;i++)x[sa[i]]= sa[i-1]+k<n&&sa[i]+k<n&&y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k] ?   p-1 : p++;if(p>=n)break;m=p;}}void get_h(){for(int i=2;i<=n;i++)lg[i]=lg[i>>1]+1;for(int i=0;i<n;i++)rk[sa[i]]=i;for(int k=0,i=0;i<n;i++){if(k>=1)k--;if(rk[i]==0)continue;int x=sa[rk[i]-1];while(s[x+k]==s[i+k])k++;h[rk[i]]=k;}memset(f,127,sizeof(f));for(int i=0;i<n;i++)f[i][0]=h[i];for(int i=1;i<=17;i++){for(int j=0;j<n;j++){f[j][i]=min(f[j][i-1],f[j+(1<<i-1)][i-1]);}}for(int i=0;i<n;i++)sum[i]=sum[i-1]+n-sa[i]-h[i];}int getk(LL k){int l=0,r=n-1;while(l<r){int mid=l+r>>1;if(sum[mid]>=k)r=mid;else l=mid+1;}return l;}int rmq(int a,int b){a=rk[a],b=rk[b];if(a>b)swap(a,b);if(a==b)return n-sa[a];a++;int len=lg[b-a+1];return min(f[a][len],f[b-(1<<len)+1][len]);}}A,B;int n,Q;char s[maxn];void solve(LL a,LL b){int la,ra,lb,rb,pa,pb,x,y;if(a>b)swap(a,b);if(b>A.sum[n-1]){puts("-1");return;}pa=A.getk(a),pb=A.getk(b);la=A.sa[pa],lb=A.sa[pb];ra=a-(pa==0 ? 0 : A.sum[pa-1]) + A.sa[pa] + A.h[pa] - 1;rb=b-(pb==0 ? 0 : A.sum[pb-1]) + A.sa[pb] + A.h[pb] - 1;x=A.rmq(la,lb),x=min(x,min(ra-la+1,rb-lb+1));y=B.rmq(n-rb-1,n-ra-1),y=min(y,min(ra-la+1,rb-lb+1));printf("%lld\n",(LL)x*x+(LL)y*y);}int main(){scanf("%d%d%s",&n,&Q,s);A.n=B.n=n;for(int i=0;i<n;i++)A.s[i]=B.s[n-i-1]=s[i];A.build(220);B.build(220);A.get_h();B.get_h();LL a,b;while(Q--){scanf("%lld%lld",&a,&b);solve(a,b);} return 0;}