BZOJ3796 Mushroom追妹纸(二分答案+后缀数组+KMP)

求出一个串使得这个串是\(s1,s2\)的子串。串中不包含\(s3\)。

如果没有这个\(s3\)就可以二分答案,然后height小于二分值分一组。看看每组里是不是出现过\(s1,s2\)的后缀。判断就行。

然后有了\(s3\)之后,我们考虑改变一下height数组。

我们把\(s1s2\)拼在一起构成一个新串\(s\)。(中间隔一个#)

设\(s3\)的长度为\(len\)。

显然对于s中出现\(s3\)的起始位置\(x\)。\(height[rk[x]]\)要小于\(len\),\(height[rk[x-1]\)要小于\(len+1\),\(height[rk[x-2]]\)要小于\(len+2\)...

这样复杂度可能是\(O(n^2)\)的,其实我们只需要把初值为INF的\(g[x]\)设为\(len\),然后倒着做一遍\(g[i]=min(g[i],g[i+1])\),\(height[rk[i]]=min(height[rk[i]],g[i])\)就行。

所以我们需要知道\(s\)中\(s3\)的起始位置。这个用KMP解决。

然后愉快的二分就行了。

#include<iostream>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<cstdio>
using namespace std;
const int N=201010;
const int INF=1e9;
int sa[N],n,m,c[N],x[N],y[N],rk[N],height[N];
int n1,n2,n3,nxt[N],g[N],ans;
char s1[N],s2[N],s3[N],s[N];
void get_sa(){
for(int i=1;i<=n;i++)c[x[i]=s[i]]++;
for(int i=1;i<=m;i++)c[i]+=c[i-1];
for(int i=n;i>=1;i--)sa[c[x[i]]--]=i;
for(int k=1;k<=n;k<<=1){
int num=0;
for(int i=n-k+1;i<=n;i++)y[++num]=i;
for(int i=1;i<=n;i++)if(sa[i]>k)y[++num]=sa[i]-k;
for(int i=1;i<=m;i++)c[i]=0;
for(int i=1;i<=n;i++)c[x[i]]++;
for(int i=1;i<=m;i++)c[i]+=c[i-1];
for(int i=n;i>=1;i--)sa[c[x[y[i]]]--]=y[i],y[i]=0;
for(int i=1;i<=n;i++)swap(x[i],y[i]);
x[sa[1]]=1;num=1;
for(int i=2;i<=n;i++)
x[sa[i]]=(y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k])?num:++num;
if(num==n)break;
m=num;
}
}
void get_height(){
int k=0;
for(int i=1;i<=n;i++)rk[sa[i]]=i;
for(int i=1;i<=n;i++){
if(rk[i]==1)continue;
if(k)k--;
int j=sa[rk[i]-1];
while(i+k<=n&&j+k<=n&&s[j+k]==s[i+k])k++;
height[rk[i]]=k;
}
for(int i=1;i<=n;i++)height[rk[i]]=min(height[rk[i]],g[i]);
}
bool judge(int x){
bool flag1=false,flag2=false;
for(int i=1;i<=n;i++){
if(height[i]<x){
flag1=flag2=false;
if(sa[i]<=n1)flag1=true;
else flag2=true;
}
else{
if(sa[i]<=n1)flag1=true;
else flag2=true;
if(flag1&&flag2)return true;
}
}
return false;
}
int main(){
scanf("%s",s1+1);
scanf("%s",s2+1);
n1=strlen(s1+1);
n2=strlen(s2+1);
for(int i=1;i<=n1;i++)s[i]=s1[i];
s[n1+1]='#';
for(int i=1;i<=n2;i++)s[n1+i+1]=s2[i];
n=n1+n2+1;
scanf("%s",s3+1);
n3=strlen(s3+1);
for(int i=2,j=0;i<=n3;i++){
while(j&&s3[j+1]!=s3[i])j=nxt[j];
if(s3[j+1]==s3[i])j++;
nxt[i]=j;
}
for(int i=1;i<=n;i++)g[i]=INF;
for(int i=1,j=0;i<=n;i++){
while(j&&(s3[j+1]!=s[i]||j==n3))j=nxt[j];
if(s3[j+1]==s[i])j++;
if(j==n3)g[i-n3+1]=n3-1;
}
for(int i=n;i>=1;i--)g[i]=min(g[i],g[i+1]+1);
m=122;get_sa();get_height();
int l=1,r=n;
while(l<=r){
int mid=(l+r)>>1;
if(judge(mid)){
ans=mid;
l=mid+1;
}
else r=mid-1;
}
printf("%d",ans);
return 0;
}
上一篇:BZOJ3796 Mushroom追妹纸 字符串 SA KMP


下一篇:【BZOJ3796】Mushroom追妹纸 二分+hash