题目描述
如果一个字符串可以被拆分为 AABB 的形式,其中 A 和 B 是任意非空字符串,则我们称该字符串的这种拆分是优秀的。
例如,对于字符串 aabaabaa,如果令 A=aab,B=a,我们就找到了这个字符串拆分成 AABB 的一种方式。
一个字符串可能没有优秀的拆分,也可能存在不止一种优秀的拆分。比如我们令 A=a,B=baa,也可以用 AABB 表示出上述字符串;但是,字符串 abaabaa 就没有优秀的拆分。
现在给出一个长度为 n 的字符串 S,我们需要求出,在它所有子串的所有拆分方式中,优秀拆分的总个数。这里的子串是指字符串中连续的一段。
以下事项需要注意:
- 出现在不同位置的相同子串,我们认为是不同的子串,它们的优秀拆分均会被记入答案。
- 在一个拆分中,允许出现 A=B。例如 cccc 存在拆分 A=B=c。
- 字符串本身也是它的一个子串。
输入格式
每个输入文件包含多组数据。输入文件的第一行只有一个整数 T,表示数据的组数。保证 1≤T≤10。
接下来 T 行,每行包含一个仅由英文小写字母构成的字符串 S,意义如题所述。
输出格式
输出 T 行,每行包含一个整数,表示字符串 S 所有子串的所有拆分中,总共有多少个是优秀的拆分。
样例一
input
4 aabbbb cccccc aabaabaabaa bbaabaababaaba
output
3 5 4 7
explanation
我们用 S[i,j] 表示字符串 SS 第 i 个字符到第 j 个字符的子串(从 1 开始计数)。
第一组数据中,共有 3 个子串存在优秀的拆分:
S[1,4]=aabb,优秀的拆分为 A=a,B=b;
S[3,6]=bbbb,优秀的拆分为 A=b,B=b;
S[1,6] = \mathtt{aabbbb}S[1,6]=aabbbb,优秀的拆分为 A= \mathtt{a}A=a,B = \mathtt{bb}B=bb。
而剩下的子串不存在优秀的拆分,所以第一组数据的答案是 33。
第二组数据中,有两类,总共 44 个子串存在优秀的拆分:
对于子串 S[1,4] = S[2,5] = S[3,6] = \mathrm{cccc}S[1,4]=S[2,5]=S[3,6]=cccc,它们优秀的拆分相同,均为 A = \mathrm{c}A=c,B = \mathrm{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。
样例二
见样例数据下载。
样例三
见样例数据下载。
限制与约定
对于全部的测试点,保证 1≤T≤10。以下对数据的限制均是对于单组输入数据而言的,也就是说同一个测试点下的 T 组数据均满足限制条件。
我们假定 n 为字符串 S 的长度,每个测试点的详细数据范围见下表:
测试点编号 | n | 其他约束 |
---|---|---|
1、2 | ≤300 | S中所有字符全部相同 |
3、4 | ≤2000 | |
5、6 | \leq 10≤10 | 无 |
7、8 | \leq 20≤20 | |
9、10 | \leq 30≤30 | |
11、12 | \leq 50≤50 | |
13、14 | \leq 100≤100 | |
15 | ≤200 | |
16 | ≤300 | |
17 | ≤500 | |
18 | ≤1000 | |
19 | ≤2000 | |
20 | ≤30000 |
时间限制:1.5s
空间限制:512MB
下载
分析
CCF送80分,然后剩下的各种算法只相差5分。
直接来看满分算法。
我们将AA和BB分开看,计算一个字符开始有多少个AA。考场上我SB了,考完才想到可以像【后缀数组】[UVA10829]L-Gap substring 那样做。
枚举A的长度l,然后把整个字符串按照长度l分段,计算si∗l和s(i+1)∗l的LCP和LCS,如果长度加起来≥l,就说明这里有AA,在相应位置计数器增加1,即cnti∗l−LCS+1,cnti∗l−LCS+1…cnti∗l−LCS+1+(LCS+LCP−1−l)都增加1。
然后再做一遍,这次不增加计数器了,增加可能的末尾位置的计数器的和,即ans+=∑(i+1)∗l+LCP−1j=(i+1)∗l+LCP−1−(LCS+LCP−l)cntj。
代码
#include<cstdio>
#include<algorithm>
#include<cstring>
#define MAXN 300000
#define MAXLOG 19
#define MAXC 256
using namespace std;
template<class T>
void Read(T &x){
static char c;
bool f(0);
while(c=getchar(),c!=EOF){
if(c=='-')
f=1;
if(c>='0'&&c<='9'){
x=c-'0';
while(c=getchar(),c>='0'&&c<='9')
x=x*10+c-'0';
ungetc(c,stdin);
if(f)
x=-x;
return;
}
}
}
int T,n,array[4][MAXN*2+10],*sa,*nsa,*rk,*nrk,b[MAXN*2+10],hn,m,st[MAXN*2+10][MAXLOG+1],height[MAXN*2+10],Log,cnt[MAXN+10],lg2[MAXN*2+10];
char s[MAXN+10];
long long ans;
void Get_sa(){
int i,k;
sa=array[0],nsa=array[1],rk=array[2],nrk=array[3];
rk[n]=nrk[n]=-1;
for(i=0;i<=MAXC;i++)
b[i]=0;
for(i=0;i<n;i++)
b[s[i]]++;
for(i=1;i<=MAXC;i++)
b[i]+=b[i-1];
for(i=0;i<n;i++)
sa[--b[s[i]]]=i;
for(rk[sa[0]]=0,i=1;i<n;i++){
rk[sa[i]]=rk[sa[i-1]];
if(s[sa[i]]!=s[sa[i-1]])
rk[sa[i]]++;
}
for(k=1;rk[sa[n-1]]<n-1;k<<=1){
for(i=0;i<n;i++)
b[rk[sa[i]]]=i;
for(i=n-1;i>=0;i--)
if(sa[i]>=k)
nsa[b[rk[sa[i]-k]]--]=sa[i]-k;
for(i=n-k;i<n;i++)
nsa[b[rk[i]]--]=i;
for(nrk[nsa[0]]=0,i=1;i<n;i++){
nrk[nsa[i]]=nrk[nsa[i-1]];
if(rk[nsa[i-1]]!=rk[nsa[i]]||rk[nsa[i-1]+k]!=rk[nsa[i]+k])
nrk[nsa[i]]++;
}
swap(rk,nrk);
swap(sa,nsa);
}
}
void Get_height(){
int i,j,k=0;
for(i=0;i<n;i++){
if(!rk[i])
height[rk[i]]=0;
else{
if(k)
k--;
for(j=sa[rk[i]-1];s[i+k]==s[j+k];k++);
height[rk[i]]=k;
}
}
}
void read(){
scanf("%s",s);
n=strlen(s);
int i;
s[n]='$';
for(i=1;i<=n;i++)
s[n+i]=s[n-i];
hn=n,n=n*2+1;
for(Log=1;(1<<Log)<=n;Log++);
Log--;
s[n]=0;
}
void prepare_st(){
int i,j;
for(i=0;i<n;i++)
*st[i]=height[i];
for(j=1;j<=Log;j++)
for(i=0;i<n;i++)
if(i+(1<<j)<=n)
st[i][j]=min(st[i][j-1],st[i+(1<<(j-1))][j-1]);
}
int Get_st(int i,int j){
if(i>j)
swap(i,j);
int t=lg2[j-i];
return min(st[i+1][t],st[j-(1<<t)+1][t]);
}
void solve(){
int i,t,t1,j;
for(i=1;i<=hn;i++){
for(j=0;j+i<hn;j+=i){
t=min(Get_st(rk[j+1],rk[j+i+1]),i-1);
t+=t1=min(Get_st(rk[n-j-i-1],rk[n-j-1]),i);
if(t>=i){
cnt[j-t1+1]++;
cnt[j-t1+1+t-i+1]--;
}
}
}
for(i=1;i<hn;i++)
cnt[i]+=cnt[i-1];
cnt[hn]=0;
for(i=1;i<=hn;i++)
cnt[i]+=cnt[i-1];
for(i=1;i<=hn;i++){
for(j=0;j+i<hn;j+=i){
t=min(Get_st(rk[n-j-i-1],rk[n-j-1]),i);
t+=t1=min(Get_st(rk[j+1],rk[j+i+1]),i-1);
if(t>=i)
ans+=cnt[j+i+t1+1]-cnt[j+i+t1+1-(t-i+1)];
}
}
}
void prepare(){
for(int i=2;i<=(MAXN<<1);i++)
lg2[i]=lg2[i>>1]+1;
}
int main()
{
prepare();
Read(T);
while(T--){
memset(cnt,0,sizeof cnt);
ans=0;
read();
Get_sa();
Get_height();
prepare_st();
solve();
printf("%lld\n",ans);
}
}