题解:
后缀数组
用height数组求出前i个后缀本质不同的子串个数
在这个数组中二分可以得到每次询问的俩个子串在原串中的位置
然后就是求这俩子串的最长公共前缀/最长公共后缀
st表解决(为了这一题我还看了一下st表)
ps:注意long long
代码:
#include<bits/stdc++.h> typedef long long ll; using namespace std; const int N=100010; int n,m,Log[N]; char ch[N]; struct SA { int p,q,k,sa[2][N],rk[2][N],mn[17][N],a[N],h[N],v[N];ll s[N]; SA(){p=0,q=1;} void mul(int *sa,int *rk,int *SA,int *RK) { 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-1]+k]!=rk[SA[i]+k]); } void getsa() { 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]]!=a[sa[p][i-1]]); for (k=1;k<n;k<<=1,swap(p,q))mul(sa[p],rk[p],sa[q],rk[q]); for (int k=0,i=1;i<=n;i++) { int j=sa[p][rk[p][i]-1]; while (a[i+k]==a[j+k])k++; h[rk[p][i]]=k;if(k)k--; } } void pre() { for (int i=1;i<=n;i++)a[i]=ch[i]-'a'+1; getsa(); for (int i=1;i<=n;i++)mn[0][i]=h[i]; for (int i=1;i<=Log[n];i++) for (int j=1;j<=n;j++) if (j+(1<<i)-1<=n)mn[i][j]=min(mn[i-1][j],mn[i-1][j+(1<<(i-1))]); else break; for (int i=1;i<=n;i++)s[i]=s[i-1]+n-sa[p][i]+1-h[i]; } int query(int a,int b) { a=rk[p][a],b=rk[p][b]; if (a>b)swap(a,b); a++; int t=Log[b-a+1]; return min(mn[t][a],mn[t][b-(1<<t)+1]); } void print() { for (int i=2;i<=n;i++)printf("%d ",h[i]); } }A,B; int main() { Log[0]=-1; for(int i=1;i<N;i++)Log[i]=Log[i>>1]+1; scanf("%lld%lld",&n,&m); scanf("%s",ch+1); A.pre(); reverse(ch+1,ch+n+1); B.pre(); for (int i=1;i<=m;i++) { ll l,r,id,a1,a2,b1,b2,ans=0; scanf("%lld%lld",&l,&r); if (l>A.s[n]||r>A.s[n]) { puts("-1"); continue; } id=lower_bound(A.s+1,A.s+n+1,l)-A.s; a1=A.sa[A.p][id]; b1=A.sa[A.p][id]+A.h[id]-1+l-A.s[id-1]; id=lower_bound(A.s+1,A.s+n+1,r)-A.s; a2=A.sa[A.p][id]; b2=A.sa[A.p][id]+A.h[id]-1+r-A.s[id-1]; ll t=(a1==a2)?2e9:A.query(a1,a2); t=min(t,min(b1-a1+1,b2-a2+1));ans+=t*t; t=(n-b1+1==n-b2+1)?2e9:B.query(n-b1+1,n-b2+1); t=min(t,min(b1-a1+1,b2-a2+1));ans+=t*t; printf("%lld\n",ans); } return 0; }