luogu1117 优秀的拆分 (后缀数组)

考虑分别计算每个位置作为AA的末尾或者BB的开头的个数 最后乘一乘就是答案

据说是套路的计算AA的方法:

首先枚举A的长度L，然后每L个字符当做一个关键点，这样的话，一个AA包含且只包含相邻两个关键点(记为a,b)，而且满足lcp(a,b)+lcs(a,b)-1>=L 手画一下就能看出来

于是SA搞lcp 倒过来再SA搞lcs 最后差分一下统计答案即可

 #include<bits/stdc++.h>
 #define pa pair<int,int>
 #define CLR(a,x) memset(a,x,sizeof(a))
 #define MP make_pair
 using namespace std;
 typedef long long ll;
 const int maxn=3e4+;
 
 inline char gc(){
     return getchar();
     static const int maxs=<<;static char buf[maxs],*p1=buf,*p2=buf;
     return p1==p2&&(p2=(p1=buf)+fread(buf,,maxs,stdin),p1==p2)?EOF:*p1++;
 }
 inline ll rd(){
     ll x=;char c=gc();bool neg=;
     while(c<''||c>''){if(c=='-') neg=;c=gc();}
     while(c>=''&&c<='') x=(x<<)+(x<<)+c-'',c=gc();
     return neg?(~x+):x;
 }
 
 char str[maxn];
 int lg2[maxn],N;
 
 struct SA{
     int cnt[maxn*],tmp[maxn*],rnk[maxn*],sa[maxn*],hei[maxn];
     int st[maxn][],n;
 
     inline void build(char *s){
         int i,j=,k,m=;
         CLR(cnt,);CLR(rnk,);
         for(i=;i<=n;i++) cnt[s[i]-'a']=;
         for(i=;i<=m;i++) cnt[i]+=cnt[i-];
         for(i=;i<=n;i++) rnk[i]=cnt[s[i]-'a'];
         for(k=;j!=n;k<<=){
             CLR(cnt,);
             for(i=;i<=n;i++) cnt[rnk[i+k]]++;
             for(i=;i<=m;i++) cnt[i]+=cnt[i-];
             for(i=n;i;i--) tmp[cnt[rnk[i+k]]--]=i;
             CLR(cnt,);
             for(i=;i<=n;i++) cnt[rnk[i]]++;
             for(i=;i<=m;i++) cnt[i]+=cnt[i-];
             for(i=n;i;i--) sa[cnt[rnk[tmp[i]]]--]=tmp[i];
             memcpy(tmp,rnk,sizeof(rnk));
             rnk[sa[]]=j=;
             for(i=;i<=n;i++){
                 if(tmp[sa[i]]!=tmp[sa[i-]]||tmp[sa[i]+k]!=tmp[sa[i-]+k]) j++;
                 rnk[sa[i]]=j;
             }m=j;
         }
 
         hei[]=;
         for(i=,j=;i<=n;i++){
             if(rnk[i]==) continue;
             if(j) j--;
             int x=sa[rnk[i]-];
             while(i+j<=n&&x+j<=n&&s[i+j]==s[x+j]) j++;
             hei[rnk[i]]=j;
         }
         // for(i=1;i<=n;i++) printf("hei:%d %d ; rnk:%d ; sa:%d \n",i,hei[i],rnk[i],sa[i]);
         for(i=n;i;i--){
             st[i][]=hei[i];
             for(j=;i+(<<j)-<=n;j++){
                 st[i][j]=min(st[i][j-],st[i+(<<(j-))][j-]);
             }
         }
     }
 
     inline int query(int x,int y){ //x,y:pos
         if(x==y) return n-y+;
         int l=min(rnk[x],rnk[y])+,r=max(rnk[x],rnk[y]);
         // printf("~%d %d\n",l,r);
         int k=lg2[r-l+];
         return min(st[l][k],st[r-(<<k)+][k]);
     }
 }fw,bw;
 
 int end[maxn],beg[maxn];
 
 int main(){
     //freopen("","r",stdin);
     int i,j,k;
     for(i=;i<=;i++) lg2[i]=lg2[i>>]+;
     for(int T=rd();T;T--){
         scanf("%s",str+);N=strlen(str+);
         fw.n=bw.n=N;
         fw.build(str);
         reverse(str+,str+N+);
         bw.build(str);
         CLR(end,);CLR(beg,);
         for(int l=;l<=N;l++){
             i=l+;
             for(;i<=N;i+=l){
                 int lcp=min(l,fw.query(i-l,i)),lcs=min(l,bw.query(N-i+,N-(i-l)+));
                 // printf("!%d %d %d %d\n",i,i-l,lcp,lcs);
                 if(lcp+lcs->=l){
                     end[i+l-lcs]++,end[i+lcp]--;
                     // printf("add A:%d %d\n",i+l-lcs,i+lcp-1);
                     beg[i-l-lcs+]++,beg[i-*l+lcp+]--;
                     // printf("add B:%d %d\n",i-l-lcs+1,i-2*l+lcp);
                 }
             }
         }
         for(i=;i<=N;i++) end[i]+=end[i-],beg[i]+=beg[i-];
         ll ans=;
         for(i=;i<N;i++){
             ans+=end[i]*beg[i+];
         }
         printf("%lld\n",ans);
     }
     return ;
 }