字符串（后缀自动机）：NOI 2016 优秀的拆分

【问题描述】

如果一个字符串可以被拆分为 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，表示数据的组数。保证 1 ≤ T ≤ 10。

接下来 T 行，每行包含一个仅由英文小写字母构成的字符串 S，意义如题所述。

【输出格式】

输出 T 行，每行包含一个整数，表示字符串 S 所有子串的所有拆分中，总共有多少个是优秀的拆分。

【样例输入】

4

aabbbb

cccccc

aabaabaabaa

bbaabaababaaba

【样例输出】

3 5 4 7

【样例说明】

我们用 S[i, j] 表示字符串 S 第 i 个字符到第 j 个字符的子串（从 1 开始计数）。

第一组数据中，共有 3 个子串存在优秀的拆分：

S[1,4] = aabb，优秀的拆分为 A = a， B = b；

S[3,6] = bbbb，优秀的拆分为 A = b， B = b；

S[1,6] = aabbbb，优秀的拆分为 A = a， B = bb。

而剩下的子串不存在优秀的拆分，所以第一组数据的答案是 3。

第二组数据中，有两类，总共 4 个子串存在优秀的拆分：

对于子串 S[1,4] = S[2,5] = S[3,6] = cccc，它们优秀的拆分相同，均为 A = c， B = c，但由于这些子串位置不同，因此要计算 3 次；

对于子串 S[1,6] = cccccc，它优秀的拆分有 2 种： A = c， B = cc 和 A = cc， B = c，它们是相同子串的不同拆分，也都要计入答案。

所以第二组数据的答案是 3 + 2 = 5。

第三组数据中， S[1,8] 和 S[4,11] 各有 2 种优秀的拆分，其中 S[1,8] 是问题描述中的例子，所以答案是 2 + 2 = 4。

第四组数据中， S[1,4]， S[6,11]， S[7,12]， S[2,11]， S[1,8] 各有 1 种优秀的拆分， S[3,14] 有 2 种优秀的拆分，所以答案是 5 + 2 = 7。

【更多样例】

下载

【样例 2 输入输出】

见目录下的 excellent/excellent2.in 与 excellent/excellent2.ans。

【样例 3 输入输出】

见目录下的 excellent/excellent3.in 与 excellent/excellent3.ans。

【子任务】

对于全部的测试点，保证 1 ≤ T ≤ 10。以下对数据的限制均是对于单组输入数据而言的，也就是说同一个测试点下的 T 组数据均满足限制条件。

我们假定 n 为字符串 S 的长度，每个测试点的详细数据范围见下表：

【来源】

NOI2016 Day1 T1

　　这道题有点难想到正解。

　　枚举长度i，然后把字符串拆分成许多连续的长度为i的子串，通过比较LCS与LCP得出一段的答案，这里发现答案是区间加法，考虑用线段树很可能超时，这里用的是差分，看程序很好理解。还有一个地方要注意：更新答案时可能会重复计算，只需要确保每次枚举都只在一段限定的区间更新，就不会出现重叠。

 #include <iostream>

 #include <cstring>

 #include <cstdio>

 using namespace std;

 const int N=;

 struct SAM{

     char s[N];

     int fa[N],pos[N],sa[N],rank[N];

     int son[N][],end[N],rht[N],lcp[N];

     int ch[N][],len[N],id[N],tot;

     int od[N],wv[N],lst,cnt;

     int mm[N],Min[N][];

     void Init(){

         memset(s,,sizeof(s));

         memset(ch,,sizeof(ch));

         memset(end,,sizeof(end));

         memset(son,,sizeof(son));

         memset(pos,,sizeof(pos));

         lst=cnt=;tot=;

     }

     void Insert(int c){

         int p=lst,np=lst=++cnt;end[lst]=;

         id[len[np]=len[p]+]=np;rht[np]=;

         while(p&&!ch[p][c])ch[p][c]=np,p=fa[p];

         if(!p)fa[np]=;

         else{

             int q=ch[p][c],nq;

             if(len[q]==len[p]+)fa[np]=q;

             else{

                 len[nq=++cnt]=len[p]+;

                 fa[nq]=fa[q];fa[q]=fa[np]=nq;

                 memcpy(ch[nq],ch[q],sizeof(ch[q]));

                 while(ch[p][c]==q)ch[p][c]=nq,p=fa[p];

             }

         }

     }

     void Get_Right(){

         for(int i=;i<=cnt;i++)wv[len[i]]++;

         for(int i=;i<=cnt;i++)wv[i]+=wv[i-];

         for(int i=;i<=cnt;i++)od[wv[len[i]]--]=i;

         for(int i=cnt;i>=;i--)rht[fa[od[i]]]+=rht[od[i]];

     }

     void Build_Tree(){

         int l=strlen(s+);

         for(int i=l;i>=;i--)Insert(s[i]-'a');

         for(int i=l;i>=;i--)

             for(int x=id[i],p=l+;x&&!pos[x];x=fa[x])

                 p-=len[x]-len[fa[x]],pos[x]=p;

         for(int x=;x<=cnt;x++)son[fa[x]][s[pos[x]]-'a']=x;

     }

     void DFS(int x,int l){

         if(end[x])sa[rank[l-len[x]+]=++tot]=l-len[x]+;

         for(int i=;i<;i++)if(son[x][i])DFS(son[x][i],l);

     }

     void Build_SA(){

         int l=strlen(s+),k=;DFS(,l);

         for(int i=,j;i<=l;lcp[rank[i++]]=k)

             for(k?k--:k,j=sa[rank[i]-];s[i+k]==s[j+k];k++);

         mm[]=-;

         for(int i=;i<=l;i++){

             mm[i]=(i&(i-))?mm[i-]:mm[i-]+;

             Min[i][]=lcp[i];

         }

         for(int k=;k<=mm[l];k++)

             for(int i=;i+(<<k-)<=l;i++)

                 Min[i][k]=min(Min[i][k-],Min[i+(<<(k-))][k-]);

     }

     int LCP(int x,int y){

         if(x>y)swap(x,y);x+=;int k=mm[y-x+];

         int ret=min(Min[x][k],Min[y-(<<k)+][k]);

         return ret;

     }

 }A,B;

 int ln,T,f[N],g[N];char s[N];

 int Get_LCP(int x,int y){return A.LCP(A.rank[x],A.rank[y]);}

 int Get_LCS(int x,int y){return B.LCP(B.rank[ln-x+],B.rank[ln-y+]);} 

 int main(){

     freopen("excellent.in","r",stdin);

     freopen("excellent.out","w",stdout);

     scanf("%d",&T);

     while(T--){

         A.Init();B.Init();

         scanf("%s",s+);ln=strlen(s+);

         for(int i=;i<=ln;i++){A.s[i]=s[i];B.s[i]=s[ln-i+];}

         A.Build_Tree();A.Build_SA();

         B.Build_Tree();B.Build_SA();

         for(int i=;i<=ln;i++)f[i]=g[i]=;

         for(int i=,l,r,x;i+i<=ln;i++)

             for(int j=i;(x=j+i)<=ln;j+=i)if(s[j]==s[x]){

                 l=x-Get_LCS(j,x)+;r=x+Get_LCP(j,x)-;

                 l=max(l+i-,x);r=min(r,x+i-);

                 if(l>r)continue;

                 f[l]++,f[r+]--;

                    g[l-i-i+]++,g[r+-i-i+]--;

             }

         long long ans=;

         for(int i=;i<=ln;i++)f[i]+=f[i-],g[i]+=g[i-];

         for(int i=;i<ln;i++)ans+=f[i]*g[i+];

         printf("%lld\n",ans);

     }

     return ;

 }