[NOI2016]优秀的拆分

一、题目

点此看题

二、解法

首先转化问题，我们可以求出$a[i],b[i]$，即以$i$结束$/$开始的$AA$串的数量，这样答案就可以表示为：$\sum a[i]\times b[i+1]$

求这两个数组，可以隔距离$len$设置一个点，这样一个长为$2\times len$的$AA$串就会被相邻的两个点统计到，然后我们分情况讨论：

第一种情况，$lcp+lcs<len$（最长公共前缀和最长公共后缀），这种情况一定不存在任何一个长度为$2\times len$的$AA$串，因为如果存在的话$lcp$应该更长。

第二种情况，$lcp+lcs\ge len$，这样的话$AA$串是可以在深红区域左右横跳的，深红段的长度就是中见红绿相交的长度，可以通过差分算出$a,b$数组。

至于$lcp$和$lcs$可以用后缀数组$+st$表$O(1)$求出，还有一些细节，比如为了不算重，我们把$lcs$从前缀$l-1$和前缀$r-1$来求，具体可以参考代码。

由调和级数求和可知，时间复杂度$O(n\log n)$。

#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;
#define ll long long
const int M = 30005;
int read()
{
    int x=0,flag=1;
    char c;
    while((c=getchar())<'0' || c>'9') if(c=='-') flag=-1;
    while(c>='0' && c<='9') x=(x<<3)+(x<<1)+(c^48),c=getchar();
    return x*flag;
}
int T,lg[M],a[M],b[M];ll ans;
struct Suffix_Array
{
    int n,m,x[M],y[M],c[M],sa[M],rk[M],hi[M];
    char s[M];int dp[M][20];
    void get_sa()
    {
        n=strlen(s+1);m=122;
        memset(c,0,sizeof c);
        memset(x,0,sizeof x);
        memset(y,0,sizeof y);
        for(int i=1;i<=n;i++) ++c[x[i]=s[i]];
        for(int i=2;i<=m;i++) c[i]+=c[i-1];
        for(int i=n;i>=1;i--) sa[c[x[i]]--]=i;
        for(int k=1;k<=n;k<<=1)
        {
            int num=0;
            for(int i=n-k+1;i<=n;i++) y[++num]=i;
            for(int i=1;i<=n;i++) if(sa[i]>k) y[++num]=sa[i]-k;
            for(int i=1;i<=m;i++) c[i]=0;
            for(int i=1;i<=n;i++) ++c[x[i]];
            for(int i=2;i<=m;i++) c[i]+=c[i-1];
            for(int i=n;i>=1;i--) sa[c[x[y[i]]]--]=y[i],y[i]=0;
            swap(x,y);
            x[sa[1]]=1;num=1;
            for(int i=2;i<=n;i++)
                x[sa[i]]=(y[sa[i]]==y[sa[i-1]] && y[sa[i]+k]==y[sa[i-1]+k])?num:++num;
            if(num==n) break;
            m=num;
        }
    }
    void get_hi()
    {
        int k=0;
        for(int i=1;i<=n;i++) rk[sa[i]]=i;
        for(int i=1;i<=n;i++)
        {
            if(rk[i]==1) continue;
            if(k) k--;
            int j=sa[rk[i]-1];
            while(j+k<=n && i+k<=n && s[i+k]==s[j+k])
                k++;
            hi[rk[i]]=k;
        }
        for(int i=1;i<=n;i++)
            dp[i][0]=hi[i];
        for(int j=1;(1<<j)<=n;j++)
            for(int i=1;i+(1<<j)-1<=n;i++)
                dp[i][j]=min(dp[i][j-1],dp[i+(1<<(j-1))][j-1]);
    }
    int ask(int l,int r)
    {
        l=rk[l];r=rk[r];
        if(l>r) swap(l,r);l++;
        int len=r-l+1,k=lg[len];
        return min(dp[l][k],dp[r-(1<<k)+1][k]);
    }
}A,B;
int main()
{
    for(int i=2;i<M;i++)
        lg[i]=lg[i>>1]+1;
    T=read();
    while(T--)
    {
        scanf("%s",A.s+1);
        int n=strlen(A.s+1);
        for(int i=1;i<=n;i++)
            B.s[n-i+1]=A.s[i];
        A.get_sa();A.get_hi();
        B.get_sa();B.get_hi();
        memset(a,0,sizeof a);
        memset(b,0,sizeof b);
        for(int len=1;len<=n/2;len++)
            for(int i=len*2;i<=n;i+=len)
            {
                int l=i-len,r=i;
                int L=n-(r-1)+1,R=n-(l-1)+1;
                int lcp=min(A.ask(l,r),len);
                int lcs=min(B.ask(L,R),len-1);
                if(lcp+lcs>=len)
                {
                    b[l-lcs]++;b[l-lcs+(lcp+lcs-len+1)]--;
                    a[r+lcp-(lcp+lcs-len+1)]++;a[r+lcp]--;
                }
            }
        ans=0;
        for(int i=1;i<=n;i++) a[i]+=a[i-1],b[i]+=b[i-1];
        for(int i=1;i<n;i++) ans+=a[i]*b[i+1];
        printf("%lld\n",ans);
    }
}