poj 2774 Long Long Message (後綴數組+二分)

題目大意：求兩個字符串的最長公共子串長度

把兩個串接在一起，中間放一個#，然後求出height

接下來是老套路，二分出一個答案ans，然後去驗證，如果有連續幾個位置的h[i]>=ans，且存在sa[i]的最大值在第二個串裏，最小值在第一個串裏，說明答案成立

別再把後綴數組敲錯了

#include <cmath>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define ll long long 
#define ull unsigned long long 
#define inf 0x3f3f3f3f
#define N 205000
using namespace std;
//re
int n,l1,l2,len;
char s1[N],s2[N],str[N];
int tr[N],rk[N],sa[N],hs[N],h[N];
int gint()
{
    int rett=0,fh=1;char c=getchar();
    while(c<'0'||c>'9'){if(c=='-')fh=-1;c=getchar();}
    while(c>='0'&&c<='9'){rett=(rett<<3)+(rett<<1)+c-'0';c=getchar();}
    return rett*fh;
}
bool check(int k,int x,int y){
    if(x+k>len||y+k>len) return 0;
    else return (rk[x]==rk[y]&&rk[x+k]==rk[y+k])?1:0;
}
void get_sa()
{
    int cnt=0,i;
    for(i=1;i<=len;i++) hs[str[i]]++;
    for(i=1;i<=127;i++) if(hs[i]) tr[i]=++cnt;
    for(i=1;i<=127;i++) hs[i]+=hs[i-1];
    for(i=1;i<=len;i++) rk[i]=tr[str[i]],sa[hs[str[i]]--]=i;
    for(int k=1;cnt<len;k<<=1)
    {
        for(i=1;i<=cnt;i++) hs[i]=0;
        for(i=1;i<=len;i++) hs[rk[i]]++;
        for(i=1;i<=cnt;i++) hs[i]+=hs[i-1];
        for(i=len;i>=1;i--) if(sa[i]>k) tr[sa[i]-k]=hs[rk[sa[i]-k]]--;
        for(i=1;i<=k;i++) tr[len-i+1]=hs[rk[len-i+1]]--;
        for(i=1;i<=len;i++) sa[tr[i]]=i;
        for(i=1,cnt=0;i<=len;i++) tr[sa[i]]=check(k,sa[i],sa[i-1])?cnt:++cnt;
        for(i=1;i<=len;i++) rk[i]=tr[i];
    }
}
void get_height()
{
    for(int i=1;i<=len;i++){
        if(rk[i]==1) continue;
        for(int j=max(1,h[rk[i-1]]-1);;j++)
            if(str[i+j-1]==str[sa[rk[i]-1]+j-1]) h[rk[i]]=j;
            else break;
    }
}
int check(int ans)
{
    for(int i=1;i<=len;){
        if(h[i]<ans){i++;continue;}
        int mi=sa[i-1],ma=sa[i-1];
        for(;i<=len&&h[i]>=ans;i++)
            mi=min(mi,sa[i]),ma=max(ma,sa[i]);
        if(mi<=l1&&ma>l1+1) return 1;
    }return 0;
}

int main()
{
    scanf("%s",s1+1);
    scanf("%s",s2+1);
    l1=strlen(s1+1),l2=strlen(s2+1);
    for(int i=1;i<=l1;i++)
        str[i]=s1[i];
    str[l1+1]='#';
    for(int i=1;i<=l2;i++)
        str[i+l1+1]=s2[i];
    len=l1+l2+1;
    get_sa();
    get_height();
    int l=0,r=min(l1,l2),ans;
    while(l<=r){
        int mid=(l+r)>>1;
        if(check(mid)) ans=mid,l=mid+1;
        else r=mid-1;
    }
    printf("%d\n",ans);
    return 0;
}