[bzoj4650]优秀的拆分

由于字符串是AABB的形式，枚举AA和BB中间的位置，分别考虑AA和BB的数量，乘起来sigma一下即为答案
以下考虑AA的情况（BB同理），枚举A的长度，然后按照这个长度分为若干块，那么每一个A一定可以表示成某一段的结尾+下一段的开头，同时两个A又要连续，也就是相邻两块的lcp和lcs，用差分来维护即可

#include<bits/stdc++.h>
using namespace std;
#define N 30005
#define mod 998244353
#define mid (l+r+1>>1)
int t,n,sum[N],mi[N],f1[N],f2[N];
long long ans;
char s[N];
int hash(int x,int y){
    if (y<x)return 0;
    if (!x)return sum[y];
    return (sum[y]-1LL*sum[x-1]*mi[y-x+1]%mod+mod)%mod;
}
bool pd(int x1,int y1,int x2,int y2){
    return hash(x1,y1)==hash(x2,y2);
}
void calc(int x,int y){
    int l=0,r=min(y-x,n-y),p=0;
    while (l<r)
        if (pd(x,x+mid-1,y,y+mid-1))l=mid;
        else r=mid-1;
    swap(l,p);
    r=min(x+1,y-x-1);
    while (l<r)
        if (pd(x-mid,x-1,y-mid,y-1))l=mid;
        else r=mid-1;
    l=p;
    if (l+r<y-x)return;
    f1[x-r]++;
    f1[2*x-y+l+1]--;
    f2[2*y-x-r-1]++;
    f2[y+l]--;
}
int main(){
    scanf("%d",&t);
    mi[0]=1;
    for(int i=1;i<N-4;i++)mi[i]=mi[i-1]*31LL%mod;
    while (t--){
        scanf("%s",s);
        memset(f1,0,sizeof(f1));
        memset(f2,0,sizeof(f2));
        n=strlen(s);
        sum[0]=s[0]-'a'+1;
        for(int i=1;i<n;i++)sum[i]=(sum[i-1]*31LL+s[i]-'a'+1)%mod;
        for(int i=1;i<=n;i++)
            for(int j=i-1;j+i<n;j+=i)calc(j,j+i);
        ans=0;
        for(int i=1;i<n;i++){
            f1[i]+=f1[i-1];
            f2[i]+=f2[i-1];
            ans+=1LL*f1[i]*f2[i-1];
        }
        printf("%lld
",ans);
    }
}