NOI2016优秀的拆分 后缀数组

题目描述

如果一个字符串可以被拆分为 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.字符串本身也是它的一个子串。

分析：
1.首先由题目想到枚举断点，再统计以当前点为结尾的AA串的个数和以下一位为开头的AA串的个数。
2.具体而言，可以枚举A的串长l，再令i=k*l,j=k*(l+1),分别统计出后缀i和后缀j的最长公共前缀R以及前缀i-1和j-1的最长公共后缀L，可以发现当（L+R）>=l 时，就会出现相应的一段区间满足条件，再利用查分的思想统计答案即可。细节较多，需要仔细思考。

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
const int maxn=300010;
int n;
int Log[maxn];
int pre[maxn],suf[maxn];
struct Saffix_Array{
    int t[maxn],t2[maxn],sa[maxn<<1],c[maxn];
    int height[maxn][20],Rank[maxn];
    char s[maxn];
    void init(){
        memset(sa,0,sizeof(sa));memset(s,0,sizeof(s));memset(t,0,sizeof(t));memset(t2,0,sizeof(t2));
        memset(height,0,sizeof(height));memset(c,0,sizeof(c));memset(Rank,0,sizeof(Rank));
    }
    void build_sa(int M){
        int *x=t,*y=t2;
        for(int i=1;i<=M;i++) c[i]=0;
        for(int i=1;i<=n;i++) c[x[i]=s[i]]++;
        for(int i=1;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 p=0;
            for(int i=n-k+1;i<=n;i++) y[++p]=i;
            for(int i=1;i<=n;i++) if(sa[i]>k) y[++p]=sa[i]-k;
            for(int i=1;i<=M;i++) c[i]=0;
            for(int i=1;i<=n;i++) c[x[y[i]]]++;
            for(int i=1;i<=M;i++) c[i]+=c[i-1];
            for(int i=n;i>=1;i--) sa[c[x[y[i]]]--]=y[i];
            swap(x,y);
            p=x[sa[1]]=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]?p:++p;
            }
            if(p>=n) break;
            M=p;
        }
        memcpy(Rank,x,sizeof(Rank));
        int k=0;
        for(int i=1;i<=n;i++){
            int j=sa[Rank[i]+1];
            if(k)k--;
            if(!j) continue;
            while(s[i+k]==s[j+k]) k++;
            height[Rank[i]][0]=k;
        }
        for(int j=1;j<=16;j++)
            for(int i=1;i+(1<<j)<=n;i++){
                height[i][j]=min(height[i][j-1],height[i+(1<<(j-1))][j-1]);
            }
    }
    int LCP(int x,int y){
        x=Rank[x],y=Rank[y];
        if(x>y) swap(x,y);
        int t=Log[y-x];
        return min(height[x][t],height[y-(1<<t)][t]);
    }
}A,B;
char s[maxn];
int main(){
    for(int i=2;i<maxn;i++) Log[i]=Log[i>>1]+1;
    int T;
    scanf("%d",&T);
    while(T--){
        A.init();B.init();
        memset(pre,0,sizeof(pre));memset(suf,0,sizeof(suf));
        scanf("%s",s+1);
        n=strlen(s+1);
        for(int i=1;i<=n;i++) A.s[i]=B.s[n+1-i]=s[i];
        A.build_sa('z');B.build_sa('z');
        for(int l=1;l<=n/2;l++){
            for(int i=l,j=i+l;j<=n;i+=l,j+=l){
                int R=min(A.LCP(i,j),l),L=min(B.LCP(n+1-(i-1),n+1-(j-1)),l-1);
                if(L+R>=l){
                    pre[j-L+l-1]++;pre[j+R]--;
                    suf[i-L]++;suf[i+R-l+1]--;
                }
            }
        }
        for(int i=1;i<=n;i++) pre[i]+=pre[i-1],suf[i]+=suf[i-1];
        long long ans=0;
        for(int i=1;i<=n;i++) ans+=pre[i]*suf[i+1];
        printf("%lld\n",ans);
    }
    return 0;
}

^_^