[Noi2016]优秀的拆分

来自F allDream的博客，未经允许，请勿转载，谢谢。

---

如果一个字符串可以被拆分为 AABB 的形式，其中 A和 B是任意非空字符串，则我们称该字符串的这种拆分是优秀的。

例如，对于字符串 aabaabaa，如果令 A=aab，B=a，我们就找到了这个字符串拆分成 AABB的一种方式。

一个字符串可能没有优秀的拆分，也可能存在不止一种优秀的拆分。比如我们令 A=a，B=baa，也可以用 AABB表示出上述字符串；但是，字符串 abaabaa 就没有优秀的拆分。

现在给出一个长度为 n的字符串 S，我们需要求出，在它所有子串的所有拆分方式中，优秀拆分的总个数。这里的子串是指字符串中连续的一段。

以下事项需要注意：

1.出现在不同位置的相同子串，我们认为是不同的子串，它们的优秀拆分均会被记入答案。

2.在一个拆分中，允许出现 A=B。例如 cccc 存在拆分 A=B=c。

3.字符串本身也是它的一个子串

你需要计算一个字符串的所有子串的优秀的拆分的个数之和

T(T<=10)组数据，n<=30000

这道题n^2暴力95分..

正解的话反正我是想不到...

如果知道以每个点为左／右端点构成AA的方案数，就可以计算答案

考虑求出正着反着各求出后缀数组之后，枚举长度L，然后每段长度是L把这个字符串分段

每次枚举两个端点i,j（i和j都是L的倍数,j=i+L），并且计算左端点在[i-L+1,i]内的答案。

所以求一下点i,j正着的lcp,反着的lcp，也就是从点i开始能向左右扩展到哪里，如果能扩展到的长度大等于L，那么就有答案

假设倒着时候的lcp是a,正着的是b，也就是[i-a+1,i]和[j-a+1,j]相同(显然a>0),b同理，那么有答案的左端点区间是[i-a+1,i-a+1 + (a+b-1-L) -1] 差分即可。

复杂度Tnlogn

#include<iostream>
#include<cstdio>
#include<cstring>
#define ll long long
#define MN 30000
#define MD 15
using namespace std;
inline int read()
{
    int x = ; char ch = getchar();
    while(ch < '' || ch > '') ch = getchar();
    while(ch >= '' && ch <= ''){x = x *  + ch - '';ch = getchar();}
    return x;
}
 
int n,k,lt[MN+],rt[MN+],v[MN+],Log[MN+];ll ans;
char st[MN*+],st2[MN*+];
 
void CalSa(int*SA,int*RK,int*sa,int*rk)
{
    for(int i=;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+;i<=n;++i) SA[v[rk[i]]--]=i;
    for(int i=;i<=n;++i)
        RK[SA[i]]=RK[SA[i-]]+(rk[SA[i]]!=rk[SA[i-]]||rk[SA[i]+k]!=rk[SA[i-]+k]);
}
 
void GetH(int*H,int*sa,int*rk,char*s)
{
    for(int i=,k=;i<=n;H[rk[i++]-]=k,k?--k:) if(rk[i]>)
        for(int j=sa[rk[i]-];s[j+k]==s[i+k];++k);
 
}
 
struct SuffixArray
{
    int p,q,sa[][MN+],rk[][MN+],f[MD+][MN+],F[MD+][MN+],fa[MD+][MN+],Fa[MD+][MN+];
    inline void Clear(){p=;q=;
        memset(rk,,sizeof(rk));
        memset(sa,,sizeof(sa));
    }
    void Build(char *s)
    {
        memset(v,,sizeof(int)*);
        for(int i=;i<=n;++i)++v[s[i]-'a'];
        for(int i=;i<;++i) v[i]+=v[i-];
        for(int i=;i<=n;++i) sa[q][v[s[i]-'a']--]=i;
        for(int i=;i<=n;++i)
            rk[q][sa[q][i]]=rk[q][sa[q][i-]]+(s[sa[q][i]]!=s[sa[q][i-]]);
        for(k=;k<n;k<<=)
        {
            CalSa(sa[p],rk[p],sa[q],rk[q]);
            swap(p,q);
        }
    }
    void BuildHeight(char *s)
    {
        GetH(f[],sa[q],rk[q],s);
        for(int i=;i<=n;++i) F[][i]=f[][i-];
        for(int i=;i<=MD;++i) F[i][]=f[i][]=f[i][n]=F[i][]=MN;
        for(int i=;i<n;++i) fa[][i]=i+,Fa[][i+]=i;
        for(int i=;i<=MD;++i)
            for(int j=;j<=n;++j)
                fa[i][j]=fa[i-][fa[i-][j]],
                Fa[i][j]=Fa[i-][Fa[i-][j]],
                 f[i][j]=min(f[i-][j],f[i-][fa[i-][j]]),
                 F[i][j]=min(F[i-][j],F[i-][Fa[i-][j]]);
    }
    int query(int i,int j)
    {
        i=rk[q][i],j=rk[q][j];
        if(i>j) swap(i,j);
        int L=Log[j-i];
        return min(f[L][i],F[L][j]);
    }
}P,S;
 
int main()
{
    Log[]=-;
    for(int i=;i<=MN;++i) Log[i]=Log[i>>]+;
    for(int T=read();T;--T)
    {
        scanf("%s",st+);n=strlen(st+);
        for(int i=;i<=n;++i) st2[i]=st[n-i+];
        P.Clear();S.Clear();ans=;
        P.Build(st);S.Build(st2);
        P.BuildHeight(st);
        S.BuildHeight(st2);
        memset(lt,,sizeof(lt));
        memset(rt,,sizeof(rt));
        for(int L=;L<=n;++L)
            for(int i=L,j=i+L;j<=n;i=j,j+=L)
            {
                int a=P.query(i,j),b=S.query(n-j+,n-i+);
                if(min(a,L)+min(b,L)->=L)
                {
                    int len=min(a,L)+min(b,L)--L;
                    ++lt[i-min(b,L)+],--lt[i-min(b,L)+len+],
                    ++rt[j-min(b,L)+L],--rt[j-min(b,L)+L+len+];
                }
            }
        for(int i=;i<=n;++i) lt[i]+=lt[i-],rt[i]+=rt[i-];
        for(int i=;i<n;++i) ans+=1LL*rt[i]*lt[i+];
        printf("%lld\n",ans);
    }
    return ;
}