题目描述
如果一个字符串可以被拆分为 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;
}
^_^