BZOJ 3796 Mushroom追妹纸 后缀数组+KMP

题意:链接

方法:后缀数组

解析:

这题正解肯定是先看s1,s2哪个位置能够匹配然后前缀和统计一下？

这个复杂度比较有正确性呀。

但是并不想写啊- -!

等暴力被卡住再说2333.

暴力思路就是和前几个题差不多，二分下答案，然后检查是否存在的时候直接暴力KMP。

我感觉这种方式的复杂度其实就是某种不严格的O(nlogn*常数)，好像卡住有点难？

另外:我承认我这篇代码写的无比丑陋- -!泥萌就不要再D我了。

代码:

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define N 10010
#define M 100010
using namespace std;
char s[M<<1],ss[M<<1];
char sss[M<<1];
int ne[M<<1];
int cnt[M],t0[M<<1],t1[M<<1],sa[M<<1],height[M<<1],rnk[M<<1];
int n,asdf;
void getsa(int m)
{
    int *x=t0,*y=t1;
    for(int i=0;i<m;i++)cnt[i]=0;
    for(int i=0;i<n;i++)cnt[x[i]=s[i]]++;
    for(int i=1;i<m;i++)cnt[i]+=cnt[i-1];
    for(int i=n-1;i>=0;i--)sa[--cnt[x[i]]]=i;
    for(int k=1;k<=n;k<<=1)
    {
        int p=0;
        for(int i=n-k;i<n;i++)y[p++]=i;
        for(int i=0;i<n;i++)if(sa[i]>=k)y[p++]=sa[i]-k;
        for(int i=0;i<m;i++)cnt[i]=0;
        for(int i=0;i<n;i++)cnt[x[y[i]]]++;
        for(int i=1;i<m;i++)cnt[i]+=cnt[i-1];
        for(int i=n-1;i>=0;i--)sa[--cnt[x[y[i]]]]=y[i];
        swap(x,y);
        p=1,x[sa[0]]=0;
        for(int i=1;i<n;i++)
            x[sa[i]]=y[sa[i-1]]==y[sa[i]]&&sa[i-1]+k<n&&sa[i]+k<n&&y[sa[i-1]+k]==y[sa[i]+k]?p-1:p++;
        if(p==n)break;
        m=p;
    }
}
void getheight()
{
    int k=0;
    for(int i=0;i<n;i++)rnk[sa[i]]=i;
    for(int i=0;i<n;i++)
    {
        if(k!=0)k--;
        if(!rnk[i])continue;
        int j=sa[rnk[i]-1];
        while(s[i+k]==s[j+k])k++;
        height[rnk[i]]=k;
    }
}
void getnext()
{
    int i=0,j=-1;
    ne[0]=-1;
    int len=strlen(ss);
    while(i<len)
    {
        if(j==-1||ss[i]==ss[j])
        {
            i++,j++;
            ne[i]=j;
        }else j=ne[j];
    }
}
int kmp(int x)
{
    int cnt=0;
    int i=0,j=0;
    while(i<x)
    {
        if(j==-1||ss[j]==sss[i])
        {
            i++,j++;
        }else j=ne[j];
        if(j==asdf)return 0;
    }
    return 1;
}
int belong[M<<1];
int minl;
int v[3];
int check(int x)
{
    int la=0;
    for(int i=1;i<=n;i++)
    {
        if(height[i]<x||i==n)
        {
            int flag=0;
            v[1]=v[2]=0;
            for(int j=la;j<i;j++)v[belong[sa[j]]]=1;
            if(v[1]==1&&v[2]==1)flag=1;
            if(flag)
            {
                for(int j=0;j<x;j++)
                {
                    sss[j]=s[sa[i-1]+j];
                }
                sss[x]='\0';
                if(kmp(x))return 1; 
            }
            la=i;
        }
    }
    return 0;
} 
int main()
{
    scanf("%s",s);
    int lll=strlen(s);s[lll]=';';
    minl=lll;
    for(int i=0;i<lll;i++)belong[i]=1;
    scanf("%s",s+lll+1);
    n=strlen(s);
    minl=min(minl,n-lll-1);
    for(int i=lll+1;i<n;i++)belong[i]=2;
    scanf("%s",ss);
    asdf=strlen(ss);
    getsa(256);
    getheight();
    getnext();
    int l=0,r=minl,ans=0;
    while(l<=r)
    {
        int mid=(l+r)>>1;
        if(check(mid))ans=mid,l=mid+1;
        else r=mid-1;
    }
    printf("%d\n",ans);
}