题意:如果一个字符串可以拆分为AABB的形式,其中A和B是任意非空字符串,则我们这种拆分是优秀的。求给定串的所有子串的拆分方案数之和。
N ≤ 30000 N \leq30000 N≤30000
本来是个神仙题
但明明一个 O ( n l o g n ) O(nlogn) O(nlogn)的题为啥只出 30000 30000 30000啊(而且只有一个点)……搞不懂
显然求出所有位置作为AA的开始和结束位置有多少个然后乘一下加起来
接下来是个神奇做法
枚举我们要求的长度 l e n len len,然后每隔 l e n len len记录一个关键点
这样每个长度为 2 ∗ l e n 2*len 2∗len的AA都恰好经过两个相邻关键点,我们用这两个关键点计算贡献。
用后缀数组算出它们向前和向后的最长的相同长度
如果中间没有相交
那么是没有贡献的
如果有相交,把相交这段标为
蓝
色
\textcolor{blue}{蓝色}
蓝色
我们发现A可以等于
蓝
色
\textcolor{blue}{蓝色}
蓝色+
绿
色
\textcolor{green}{绿色}
绿色+
紫
色
\textcolor{purple}{紫色}
紫色
挪到后面也可以等于 绿 色 \textcolor{green}{绿色} 绿色+ 紫 色 \textcolor{purple}{紫色} 紫色+ 蓝 色 \textcolor{blue}{蓝色} 蓝色
中间随便截一段也是合法的
用个差分即可
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cctype>
#include <algorithm>
#define MAXN 30005
using namespace std;
typedef long long ll;
int LOG[MAXN];
char s[MAXN];
int n;
struct SuffixArray
{
int sa[MAXN],rk[MAXN],tp[MAXN];
int c[MAXN],ht[MAXN];
int st[MAXN][20];
int m;
void Rsort()
{
for (int i=1;i<=m;i++) c[i]=0;
for (int i=1;i<=n;i++) ++c[rk[i]];
for (int i=1;i<=m;i++) c[i]+=c[i-1];
for (int i=n;i>=1;i--) sa[c[rk[tp[i]]]--]=tp[i];
}
void build()
{
memset(rk,0,sizeof(rk));
memset(sa,0,sizeof(sa));
memset(tp,0,sizeof(tp));
m='z';
for (int i=1;i<=n;i++) rk[i]=s[i],tp[i]=i;
Rsort();
for (int w=1,p=0;p<n;w<<=1,m=p)
{
p=0;
for (int i=n-w+1;i<=n;i++) tp[++p]=i;
for (int i=1;i<=n;i++) if (sa[i]>w) tp[++p]=sa[i]-w;
Rsort();
swap(rk,tp);
rk[sa[1]]=p=1;
for (int i=2;i<=n;i++)
if (tp[sa[i]]==tp[sa[i-1]]&&tp[sa[i]+w]==tp[sa[i-1]+w]) rk[sa[i]]=p;
else rk[sa[i]]=++p;
}
int k=1;
for (int i=1;i<=n;i++)
{
if (k) --k;
int j=sa[rk[i]-1];
while (s[i+k]==s[j+k]) ++k;
ht[rk[i]]=k;
}
memset(st,0,sizeof(st));
for (int i=1;i<=n;i++) st[i][0]=ht[i];
for (int j=1;j<20;j++)
for (int i=1;i+(1<<(j-1))<=n;i++)
st[i][j]=min(st[i][j-1],st[i+(1<<(j-1))][j-1]);
}
inline int query_st(int l,int r)
{
int t=LOG[r-l+1];
return min(st[l][t],st[r-(1<<t)+1][t]);
}
inline int query(int x,int y)
{
if (rk[x]>rk[y]) swap(x,y);
return query_st(rk[x]+1,rk[y]);
}
}A,B;
#define lcp B.query
#define lcs A.query
int p[MAXN],q[MAXN];
int main()
{
LOG[0]=-1;
for (int i=1;i<MAXN;i++) LOG[i]=LOG[i>>1]+1;
int T;
scanf("%d",&T);
while (T--)
{
scanf("%s",s+1);
n=strlen(s+1);
A.build();
reverse(s+1,s+n+1);
B.build();
for (int i=1;i<=n;i++) p[i]=q[i]=0;
for (int len=1;len<=n;len++)
for (int i=len;i+len<=n;i+=len)
{
int j=i+len;
int a=min(lcp(n-i+1,n-j+1),len),b=min(lcs(i,j),len);
if (a+b<=len) continue;
int s=a+b-len-1;
++p[i-a+1];--p[i-a+s+2];
++q[i-a+2*len];--q[i-a+s+2*len+1];
}
for (int i=1;i<=n;i++) p[i]+=p[i-1],q[i]+=q[i-1];
ll ans=0;
for (int i=1;i<n;i++) ans+=(ll)q[i]*p[i+1];
printf("%lld\n",ans);
}
return 0;
}