【后缀数组】[NOI2016]优秀的拆分
题目描述
如果一个字符串可以被拆分为
例如,对于字符串 aabaabaa,如果令
一个字符串可能没有优秀的拆分,也可能存在不止一种优秀的拆分。比如我们令
现在给出一个长度为
以下事项需要注意:
- 出现在不同位置的相同子串,我们认为是不同的子串,它们的优秀拆分均会被记入答案。
- 在一个拆分中,允许出现
A=B 。例如 cccc 存在拆分A=B=c 。 - 字符串本身也是它的一个子串。
输入格式
每个输入文件包含多组数据。输入文件的第一行只有一个整数
接下来
输出格式
输出
样例一
input
4 aabbbb cccccc aabaabaabaa bbaabaababaaba
output
3 5 4 7
explanation
我们用
第一组数据中,共有
S[1,6] = \mathtt{aabbbb}
而剩下的子串不存在优秀的拆分,所以第一组数据的答案是 3
第二组数据中,有两类,总共 4
对于子串 S[1,4] = S[2,5] = S[3,6] = \mathrm{cccc}
对于子串
所以第二组数据的答案是
第三组数据中,
第四组数据中,
样例二
见样例数据下载。
样例三
见样例数据下载。
限制与约定
对于全部的测试点,保证
我们假定
测试点编号 | 其他约束 | |
---|---|---|
1、2 | ||
3、4 | ||
5、6 | \leq 10 | 无 |
7、8 | \leq 20 | |
9、10 | \leq 30 | |
11、12 | \leq 50 | |
13、14 | \leq 100 | |
15 | ||
16 | ||
17 | ||
18 | ||
19 | ||
20 |
时间限制:
空间限制:
下载
分析
CCF送80分,然后剩下的各种算法只相差5分。
直接来看满分算法。
我们将
枚举
然后再做一遍,这次不增加计数器了,增加可能的末尾位置的计数器的和,即
代码
#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);
}
}