【BZOJ4650】【NOI2016】优秀的拆分（后缀数组）

【BZOJ4650】【NOI2016】优秀的拆分（后缀数组）

题面

BZOJ

Uoj

题解

如果我们知道以某个位置为开始/结尾的\(AA\)串的个数

那就直接做一下乘法就好

这个怎么求？

枚举一个位置

枚举串的长度

直接暴力算就好啦

至于是否可行，用\(SA\)求\(lcp\)就好啦

这样就是\(95\)分

NOI这么好拿部分分的？？？

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<set>
#include<map>
#include<vector>
#include<queue>
using namespace std;
#define ll long long
#define MAX 35000
int x[MAX],y[MAX],t[MAX];
int SA[MAX],height[MAX],rk[MAX];
int lg[MAX],n,p[20][MAX],a[MAX];
char s[MAX];
int g[MAX],f[MAX],T;
bool cmp(int i,int j,int k){return y[i]==y[j]&&y[i+k]==y[j+k];}
void init()
{
memset(SA,0,sizeof(SA));
memset(height,0,sizeof(height));
memset(rk,0,sizeof(rk));
memset(x,0,sizeof(x));
memset(y,0,sizeof(y));
memset(t,0,sizeof(t));
memset(a,0,sizeof(a));
}
void GetSA()
{
int m=50;
for(int i=1;i<=n;++i)t[x[i]=a[i]]++;
for(int i=1;i<=m;++i)t[i]+=t[i-1];
for(int i=n;i>=1;--i)SA[t[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=0;i<=m;++i)t[i]=0;
for(int i=1;i<=n;++i)t[x[y[i]]]++;
for(int i=1;i<=m;++i)t[i]+=t[i-1];
for(int i=n;i>=1;--i)SA[t[x[y[i]]]--]=y[i];
swap(x,y);
x[SA[1]]=p=1;
for(int i=2;i<=n;++i)
x[SA[i]]=cmp(SA[i],SA[i-1],k)?p:++p;
if(p>=n)break;
m=p;
}
for(int i=1;i<=n;++i)rk[SA[i]]=i;
for(int i=1,j=0;i<=n;++i)
{
if(j)--j;
while(a[i+j]==a[SA[rk[i]-1]+j])++j;
height[rk[i]]=j;
}
}
void Pre()
{
memset(p,63,sizeof(p));
for(int i=1;i<=n;++i)p[0][i]=height[i];
for(int j=1;j<15;++j)
for(int i=1;i<=n;++i)
p[j][i]=min(p[j-1][i],p[j-1][i+(1<<(j-1))]);
}
int Query(int i,int j)
{
return min(p[lg[j-i+1]][i],p[lg[j-i+1]][j-(1<<lg[j-i+1])+1]);
}
int lcp(int i,int j)
{
int l=min(rk[i],rk[j])+1,r=max(rk[i],rk[j]);
return Query(l,r);
}
int main()
{
for(int i=2;i<=30000;++i)lg[i]=lg[i>>1]+1;
scanf("%d",&T);
while(T--)
{
init();
scanf("%s",s+1);
n=strlen(s+1);
for(int i=1;i<=n;++i)a[i]=s[i]-96;
GetSA();Pre();
for(int i=1;i<=n;++i)
{
g[i]=0;
for(int l=1;l+l+i-1<=n;++l)
if(lcp(i,i+l)>=l)g[i]++;
}
for(int i=2;i<=n;++i)
{
f[i]=0;
for(int l=1;i-l-l+1>0;++l)
if(lcp(i-l-l+1,i-l+1)>=l)f[i]++;
}
int ans=0;
for(int i=1;i<n;++i)
ans+=f[i]*g[i+1];
printf("%d\n",ans);
}
return 0;
}

\(95\)分的暴力太显然了。。

原来\(NOI\)都是这样送分？？？

为什么NOIP 没有这么好的福利

想想怎么优化吧。。。

肯定不能枚举长度之后再暴力算每一个位置

那么，我们要考虑一个方法，

可以一次性算出连续的位置

想想我们怎么求\(AA\)这种形式??

计算\(lcp(i,i+len)>=len\)是否成立

但是，如果\(lcp(i,i+len)>=len\)

我们就会发现，有一段区间内都是有满足条件的子串

所以我们可以一起计算

现在仔细思考怎么算

因为每次是\(i\)和\(i+len\)

所以我们只要枚举位置是\(len\)的倍数的地方就好

旁边的地方我们要想办法算出来

第一个，是向后如果可以增加的话

\(lcp(i,i+len)>=L\)我就会获得向后的一段连续区间

如果只算向后，会忽略掉向前的一段

所以再算一下\(lcs(i,i+len)\)这段，这两边拼起来

如果满足条件，证明这一段区间都是可行的

这样就可以差分全部\(+1\)

如果重复的部分够多

这样算可能会影响到别的块里面

所以要强制只在自己这一段里面算

具体的实现看代码啦

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<set>
#include<map>
#include<vector>
#include<queue>
using namespace std;
#define ll long long
#define MAX 35000
int lg[MAX],n;
char s[MAX];
int g[MAX],f[MAX],T;
struct SA
{
int p[20][MAX],a[MAX];
int x[MAX],y[MAX],t[MAX];
int SA[MAX],height[MAX],rk[MAX];
bool cmp(int i,int j,int k){return y[i]==y[j]&&y[i+k]==y[j+k];}
void init()
{
memset(SA,0,sizeof(SA));
memset(height,0,sizeof(height));
memset(rk,0,sizeof(rk));
memset(x,0,sizeof(x));
memset(y,0,sizeof(y));
memset(t,0,sizeof(t));
memset(a,0,sizeof(a));
}
void GetSA()
{
int m=50;
for(int i=1;i<=n;++i)t[x[i]=a[i]]++;
for(int i=1;i<=m;++i)t[i]+=t[i-1];
for(int i=n;i>=1;--i)SA[t[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=0;i<=m;++i)t[i]=0;
for(int i=1;i<=n;++i)t[x[y[i]]]++;
for(int i=1;i<=m;++i)t[i]+=t[i-1];
for(int i=n;i>=1;--i)SA[t[x[y[i]]]--]=y[i];
swap(x,y);
x[SA[1]]=p=1;
for(int i=2;i<=n;++i)
x[SA[i]]=cmp(SA[i],SA[i-1],k)?p:++p;
if(p>=n)break;
m=p;
}
for(int i=1;i<=n;++i)rk[SA[i]]=i;
for(int i=1,j=0;i<=n;++i)
{
if(j)--j;
while(a[i+j]==a[SA[rk[i]-1]+j])++j;
height[rk[i]]=j;
}
}
void Pre()
{
memset(p,63,sizeof(p));
for(int i=1;i<=n;++i)p[0][i]=height[i];
for(int j=1;j<15;++j)
for(int i=1;i<=n;++i)
p[j][i]=min(p[j-1][i],p[j-1][i+(1<<(j-1))]);
}
int Query(int i,int j)
{
return min(p[lg[j-i+1]][i],p[lg[j-i+1]][j-(1<<lg[j-i+1])+1]);
}
int lcp(int i,int j)
{
int l=min(rk[i],rk[j])+1,r=max(rk[i],rk[j]);
return Query(l,r);
}
}A,B;
int main()
{
for(int i=2;i<=30000;++i)lg[i]=lg[i>>1]+1;
scanf("%d",&T);
while(T--)
{
A.init();B.init();
scanf("%s",s+1);
n=strlen(s+1);
for(int i=1;i<=n;++i)A.a[i]=s[i]-96;
for(int i=1;i<=n;++i)B.a[n-i+1]=s[i]-96;
A.GetSA();A.Pre();B.GetSA();B.Pre();
for(int i=1;i<=n;++i)g[i]=f[i]=0;
for(int len=1;len<=n/2;++len)
{
for(int i=len,j=i+len;j<=n;i+=len,j+=len)
{
int x=min(A.lcp(i,j),len);
int y=min(B.lcp(n-i+2,n-j+2),len-1);
int t=x+y-len+1;
if(x+y>=len)
{
g[i-y]++;g[i-y+t]--;
f[j+x-t]++;f[j+x]--;
}
}
}

for(int i=1;i<=n;++i)g[i]+=g[i-1];
for(int i=1;i<=n;++i)f[i]+=f[i-1];
ll ans=0;
for(int i=1;i<n;++i)
ans+=1ll*f[i]*g[i+1];
printf("%lld\n",ans);
}
return 0;
}