BZOJ3796 Mushroom追妹纸

Description

给出字符串s1、s2、s3，找出一个字符串w，满足：

1、w是s1的子串；

2、w是s2的子串；

3、s3不是w的子串。

4、w的长度应尽可能大

求w的最大长度。

---

Input

输入有三行，第一行为一个字符串s1第二行为一个字符串s2，

第三行为一个字符串s3。输入仅含小写字母，字符中间不含空格。

---

Output

输出仅有一行，为w的最大可能长度，如w不存在，则输出0。

---

Solution

要求匹配的串只有两个，可以把它们连起来求一遍SA。
然后KMP求出c在大串中出现的位置。
维护一个数组f[i]表示从i位置开始至多延长多少个位置保证不包含c。
然后扫两遍，一遍算第一个串对第二个串的贡献，再算第二个串对第一个串的贡献，滚动维护一下min(height)就好了。

---

Code

#include<bits/stdc++.h>
using namespace std;
#define N 200010
int n,m,sa[N],rnk[N],t[N],y[N],height[N],n1,fail[N],f[N],ans;
bool b[N],tag[N];
char s[N],s1[N];
void qsort(){
    for(int i=0;i<=m;++i)t[i]=0;
    for(int i=1;i<=n;++i)t[rnk[i]]++;
    for(int i=1;i<=m;++i)t[i]+=t[i-1];
    for(int i=n;i>=1;--i)sa[t[rnk[y[i]]]--]=y[i];
}
void SA(){
    m=200;
    for(int i=1;i<=n;++i)rnk[i]=s[i],y[i]=i;
    qsort();
    for(int w=1,p=0;p<n;m=p,w<<=1){
        p=0;
        for(int i=n-w+1;i<=n;++i)y[++p]=i;
        for(int i=1;i<=n;++i)if(sa[i]>w)y[++p]=sa[i]-w;
        qsort();swap(rnk,y);rnk[sa[1]]=p=1;
        for(int i=2;i<=n;++i)rnk[sa[i]]=((y[sa[i]]==y[sa[i-1]])&&(y[sa[i]+w]==y[sa[i-1]+w]))?p:++p;
    }    
    for(int i=1;i<=n;++i){
        if(rnk[i]==1)continue;
        int j=max(0,height[rnk[i-1]]-1);
        while(s[i+j]==s[sa[rnk[i]-1]+j])j++;
        height[rnk[i]]=j;
    }   
}
int main(){
    scanf("%s%s",s+1,s1+1);
    n=strlen(s+1);n1=strlen(s1+1);
    for(int i=1;i<=n;++i)f[i]=n-i+1;s[++n]='#'; 
    for(int i=1;i<=n1;++i)s[++n]=s1[i],b[n]=1,f[n]=n1-i+1;
    SA();
    scanf("%s",s1+1);n1=strlen(s1+1);
    int j=0;
    for(int i=2;i<=n1;++i){
        while(j&&s1[i]!=s1[j+1])j=fail[j];
        if(s1[i]==s1[j+1])j++;fail[i]=j;
    }
    j=0;
    for(int i=1;i<=n;++i){
        while(j&&s[i]!=s1[j+1])j=fail[j];
        if(s[i]==s1[j+1])j++;
        if(j==n1)tag[i-j+1]=1;
    }
    int xian=n+1;
    for(int i=n;i>=1;--i){
        if(tag[i])xian=min(xian,i+n1-1);
        f[i]=min(f[i],xian-i);
    }
    xian=0;
    for(int i=1;i<=n;++i){
        int pos=sa[i];
        if(!b[pos])xian=min(xian,height[i]),ans=max(ans,min(f[pos],xian));
        else xian=2e9;
    }
    xian=0;
    for(int i=1;i<=n;++i){
        int pos=sa[i];
        if(b[pos])xian=min(xian,height[i]),ans=max(ans,min(f[pos],xian));
        else xian=2e9;
    }
    printf("%d",ans);
}