bzoj3796 Mushroom追妹纸(SA+二分答案+kmp)


把s1和s2接在一起,求最长公共子串。二分答案就好了。。。此题就多一个判断,要求s3不能出现在公共子串里,那我们就预处理一下,用kmp求出s3在s2中出现的位置,处理成L数组,即可O(1)判断。

#include <cstdio>
#include <cstring>
#include <algorithm>
#define ll long long
#define N 100100
char s[50010],t[10010];
int a[N],n=0,m=30,bel[N],len[4],fail[10010],L[N];
int rank[N<<1],rank1[N],sa[N],h[N],tmp[N],count[N];
void getfail(){
    int k=0;fail[1]=0;
    for(int i=2;i<=len[3];++i){
        while(k&&t[k+1]!=t[i]) k=fail[k];
        if(t[k+1]==t[i]) ++k;
        fail[i]=k;
    }
}
void kmp(){//L[i]表示L[i]...i包含s3的最小区间 
    int k=0,last=0;
    for(int i=1;i<=len[2];++i){
        while(k&&t[k+1]!=s[i]) k=fail[k];
        if(t[k+1]==s[i]) ++k;
        if(k==len[3]) last=len[1]+1+i-len[3]+1;
        L[len[1]+1+i]=last;
    }
}
void SA(){
    for(int i=1;i<=m;++i) count[i]=0;
    for(int i=1;i<=n;++i) count[a[i]]=1;
    for(int i=1;i<=m;++i) count[i]+=count[i-1];
    for(int i=n;i>=1;--i) rank[i]=count[a[i]];
    int k=0;
    for(int p=1;k!=n;p<<=1,m=k){
        for(int i=1;i<=m;++i) count[i]=0;
        for(int i=1;i<=n;++i) count[rank[i+p]]++;
        for(int i=1;i<=m;++i) count[i]+=count[i-1];
        for(int i=n;i>=1;--i) tmp[count[rank[i+p]]--]=i;
        for(int i=1;i<=m;++i) count[i]=0;
        for(int i=1;i<=n;++i) count[rank[tmp[i]]]++;
        for(int i=1;i<=m;++i) count[i]+=count[i-1];
        for(int i=n;i>=1;--i) sa[count[rank[tmp[i]]]--]=tmp[i];
        memcpy(rank1,rank,sizeof(rank1));
        rank[sa[1]]=k=1;
        for(int i=2;i<=n;++i){
            if(rank1[sa[i]]!=rank1[sa[i-1]]||rank1[sa[i]+p]!=rank1[sa[i-1]+p]) ++k;
            rank[sa[i]]=k;
        }
    }k=0;
    for(int i=1;i<=n;++i){
        if(rank[i]==1){h[1]=0;continue;}
        if(i==1||h[rank[i-1]]<=1) k=0;
        if(k) --k;
        while(a[i+k]==a[sa[rank[i]-1]+k]) ++k;
        h[rank[i]]=k;
    }
}
inline bool jud(int x){
    bool f[3];f[0]=f[1]=f[2]=0;
    for(int i=1;i<=n+1;++i){
        if(h[i]>=x){
            if(sa[i]<=L[sa[i]+x-1]) continue;
            f[bel[sa[i]]]=1;
        }else{
            if(f[1]&&f[2]) return 1;
            f[1]=f[2]=0;
            if(sa[i]<=L[sa[i]+x-1]) continue;
            f[bel[sa[i]]]=1;
        }
    }
    return 0;
}
int main(){
// freopen("a.in","r",stdin);
    int l=1,r=0;
    for(int i=1;i<=2;++i){
        scanf("%s",s+1);len[i]=strlen(s+1);r=std::max(r,len[i]);
        for(int j=1;j<=len[i];++j) a[++n]=s[j]-'a'+1,bel[n]=i;
        a[++n]=++m;
    }
    scanf("%s",t+1);len[3]=strlen(t+1);
    getfail();
    kmp();//预处理出s3在s2中出现的位置 
    SA();
    while(l<=r){//二分答案 
        int mid=l+r>>1;
        if(jud(mid)) l=mid+1;
        else r=mid-1;
    }
    printf("%d\n",l-1);
    return 0;
}
智能推荐

注意!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系我们删除。



 
© 2014-2019 ITdaan.com 粤ICP备14056181号  

赞助商广告