poj 2774 Long Long Message

把第二个串连到第一个上，求一下后缀数组，然后处理height的时候，判断是不是当前位置和上一个位置处在原来的不同串就好了（而且不能只判断一个串内，只到n1会掉下 (i>n1 && j<=n1)这种情况。。。）

#include<cstdio>
#include<cstring>
#include<iostream>
#define N 100005<<2
#define LL long long
#define inf 0x3f3f3f3f
using namespace std;
inline int ra()
{
    int x=0,f=1; char ch=getchar();
    while (ch<'0' || ch>'9') {if (ch=='-') f=-1; ch=getchar();}
    while (ch>='0' && ch<='9') {x=x*10+ch-'0'; ch=getchar();}
    return x*f;
}
char ch[N],ch1[N],ch2[N];
int n,k,p,q,n1,n2,ans;
int a[N],rank[2][N],sa[2][N],h[N],v[N];
void cal_sa(int sa[N], int rank[N], int Sa[N], int Rank[N])
{
    for (int i=1; i<=n; i++) v[rank[sa[i]]]=i;
    for (int i=n; i>=1; i--)    
        if (sa[i]>k) Sa[v[rank[sa[i]-k]]--]=sa[i]-k;
    for (int i=n-k+1; i<=n; i++) Sa[v[rank[i]]--]=i;
    for (int i=1; i<=n; i++) Rank[Sa[i]]=Rank[Sa[i-1]]+(rank[Sa[i]]!=rank[Sa[i-1]] || rank[Sa[i]+k]!=rank[Sa[i-1]+k]);
}
void get_height()
{
    k=0;
    for (int i=1; i<=n; i++)   //只到n1会掉下 (i>n1 && j<=n1)这种情况。。。 
    { 
        if (rank[p][i]==1) h[1]=0;
        else{
            int j=sa[p][rank[p][i]-1];
            while (a[i+k]==a[j+k]) k++;
            if ((j>n1 && i<=n1) || (i>n1 && j<=n1)) ans=max(ans,k);
            h[rank[p][i]]=k; if (k>0) k--;
        }
    }
}
void work()
{
    p=0,q=1;
    for (int i=1; i<=n; i++) v[a[i]]++;
    for (int i=1; i<=30; i++) v[i]+=v[i-1];
    for (int i=1; i<=n; i++) sa[p][v[a[i]]--]=i;
    for (int i=1; i<=n; i++) rank[p][sa[p][i]]=rank[p][sa[p][i-1]]+(a[sa[p][i]]!=a[sa[p][i-1]]);
    for (k=1; k<n; k<<=1,q^=1,p^=1) cal_sa(sa[p],rank[p],sa[q],rank[q]);
    get_height();
}
int main()
{
    scanf("%s",ch1+1); scanf("%s",ch2+1);
    n1=strlen(ch1+1); n2=strlen(ch2+1); n=n1+n2;
    for (int i=1; i<=n1; i++) a[i]=ch1[i]-'a'+1;
    for (int i=n1+1; i<=n; i++) a[i]=ch2[i-n1]-'a'+1;
    work();
    cout<<ans;
    return 0;
}