传送门:bzoj4650
题解
考虑枚举对称点,设 q z i , h z i qz_i,hz_i qzi,hzi分别为以 i i i结尾/开头的形式为 A A AA AA的字符串的个数,则 a n s = ∑ i = 1 n − 1 q z i × h z i + 1 ans=\sum\limits_{i=1}^{n-1}qz_i\times hz_{i+1} ans=i=1∑n−1qzi×hzi+1
考虑一个长为 l l l的 A A AA AA串必然经过至少2个原串中 i = k × l 2 i=k\times {\dfrac l2} i=k×2l的位置,所以可以 O ( n ln n ) O(n\ln n) O(nlnn)枚举枚举 A A AA AA串中 A A A的长度 l l l以及 i = k × l , j = i + l , j ≤ n i=k\times l,j=i+l,j\leq n i=k×l,j=i+l,j≤n处对答案的贡献。
设 x = l c s ( p r e ( i − 1 ) , p r e ( j − 1 ) ) , y = l c p ( s u f ( i ) , s u f ( j ) ) x=lcs(pre(i-1),pre(j-1)),y=lcp(suf(i),suf(j)) x=lcs(pre(i−1),pre(j−1)),y=lcp(suf(i),suf(j)),若 x + y ≥ l x+y\geq l x+y≥l,则总共存在 x + y − l + 1 x+y-l+1 x+y−l+1个覆盖这两个端点长度为 2 l 2l 2l的 A A AA AA串,等价于区间 f , g f,g f,g的+1操作,差分左端点+1,右端点后-1,最后求前缀和即可。
快速求前后缀 l c p , l c s lcp,lcs lcp,lcs正反做2遍后缀数组后 s t st st表记录即可。
一些需要注意的细节:
1.计算
x
=
l
c
s
(
p
r
e
(
i
−
1
)
,
p
r
e
(
j
−
1
)
)
,
y
=
l
c
p
(
s
u
f
(
i
)
,
s
u
f
(
j
)
)
x=lcs(pre(i-1),pre(j-1)),y=lcp(suf(i),suf(j))
x=lcs(pre(i−1),pre(j−1)),y=lcp(suf(i),suf(j))有可能
x
≥
l
x\geq l
x≥l或者
y
>
l
y> l
y>l,导致枚举到的某些区间实际上并不完全覆盖这两个点,不能当前点对的计入贡献。所以
x
,
y
x,y
x,y要分别与
l
−
1
,
l
l-1,l
l−1,l取
m
i
n
min
min。
2.多组询问时记得清空
S
A
SA
SA中两个桶数组。(不清空判断y[sa[i]+k]!=y[sa[i-1]+k]
时会爆掉)。
代码
#include<bits/stdc++.h>
#define mem(f,x) memset((f),(x),sizeof((f)))
using namespace std;
typedef long long ll;
const int N=6e4+10;
int tk,n,m,bin[30],lg[N];
int qz[N],hz[N];char s[N];ll ans;
struct SA{
int m,sa[N],t1[N],t2[N],c[N],rk[N];
int ht[17][N];
inline void itia(){mem(ht,0x7f);mem(t1,0);mem(t2,0);}
inline void build()
{
int i,k,p,*x=t1,*y=t2;m=26;
for(i=1;i<=m;++i) c[i]=0;
for(i=1;i<=n;++i) c[(x[i]=(int)s[i])]++;
for(i=1;i<=m;++i) c[i]+=c[i-1];
for(i=n;i;--i) sa[c[x[i]]--]=i;
for(k=1;k<n;k<<=1){
//the second key
for(p=0,i=n-k+1;i<=n;++i) y[++p]=i;
for(i=1;i<=n;++i) if(sa[i]>k) y[++p]=sa[i]-k;
//the first key
for(i=1;i<=m;++i) c[i]=0;
for(i=1;i<=n;++i) c[x[y[i]]]++;
for(i=1;i<=m;++i) c[i]+=c[i-1];
for(i=n;i;--i) sa[c[x[y[i]]]--]=y[i];
p=1;swap(x,y);x[sa[1]]=1;
for(i=2;i<=n;++i){
p+=((y[sa[i]]!=y[sa[i-1]])||(y[sa[i]+k]!=y[sa[i-1]+k]))?1:0;
x[sa[i]]=p;
}
if(p>=n) break;
m=p;
}
}
inline void mk()
{
int i,j,k=0;
for(i=1;i<=n;++i) rk[sa[i]]=i;
for(i=1;i<=n;++i){
if(rk[i]==1) {k=0;continue;}
if(k) k--;j=sa[rk[i]-1];
for(;j+k<=n && i+k<=n && s[j+k]==s[i+k];++k);
ht[0][rk[i]]=k;
}
for(k=1;bin[k]<=n;++k){
for(i=1;i+bin[k]-1<=n;++i)
ht[k][i]=min(ht[k-1][i],ht[k-1][i+bin[k-1]]);
}
}
inline int ask(int x,int y)
{
x=rk[x];y=rk[y];if(x>y) swap(x,y);
int cg=lg[y-x];
return min(ht[cg][x+1],ht[cg][y-bin[cg]+1]);
}
}A,B;
inline void sol()
{
int i,j,x,y,z,l;
A.itia();B.itia();mem(qz,0);mem(hz,0);
scanf("%s",s+1);n=strlen(s+1);
for(i=1;i<=n;++i) s[i]=s[i]-'a'+1;
A.build();A.mk();
reverse(s+1,s+n+1);
B.build();B.mk();
for(l=1;l+l<=n;++l)
for(i=l,j=l+l;j<=n;i+=l,j+=l){
x=min(B.ask(n+2-j,n+2-i),l-1);y=min(A.ask(i,j),l);
if(x+y<l) continue;z=x+y-l+1;
qz[j+y-z]++;qz[j+y]--;
hz[i-x]++;hz[i-x+z]--;
}
for(i=1;i<=n;++i) qz[i]+=qz[i-1],hz[i]+=hz[i-1];
ans=0LL;
for(i=1;i<n;++i) ans+=(ll)qz[i]*hz[i+1];
printf("%lld\n",ans);
}
int main(){
int i;bin[0]=1;
for(i=1;i<25;++i) bin[i]=bin[i-1]<<1;
for(i=2;i<N;++i) lg[i]=lg[i>>1]+1;
for(scanf("%d",&tk);tk;--tk) sol();
return 0;
}