POJ 2774 Long Long Message（后缀数组）

题目链接：http://poj.org/problem?id=2774

题意：给定两个字符串A和B，求最长公共子串。

思路：将两个串连在一起。枚举h，如果sa[i]和sa[i-1]在连接处的两侧则该h值可以更新答案。

int r[N],sa[N],wa[N],wb[N],wd[N],rank[N],h[N];





int cmp(int *r,int a,int b,int L)

{

    return r[a]==r[b]&&r[a+L]==r[b+L];

}





void da(int *r,int *sa,int n,int m)

{

    int i,j,p,*x=wa,*y=wb,*t;

    FOR0(i,m) wd[i]=0;

    FOR0(i,n) wd[x[i]=r[i]]++;

    FOR1(i,m-1) wd[i]+=wd[i-1];

    FORL0(i,n-1) sa[--wd[x[i]]]=i;

    for(j=1,p=1;p<n;j<<=1,m=p)

    {

        p=0;

        FOR(i,n-j,n-1) y[p++]=i;

        FOR0(i,n) if(sa[i]>=j) y[p++]=sa[i]-j;

        FOR0(i,m) wd[i]=0;

        FOR0(i,n) wd[x[i]]++;

        FOR1(i,m-1) wd[i]+=wd[i-1];

        FORL0(i,n-1) sa[--wd[x[y[i]]]]=y[i];

        t=x;x=y;y=t;p=1;x[sa[0]]=0;

        FOR1(i,n-1) x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;

    }

}







void calHeight(int *r,int *sa,int n)

{

    int i,j,k=0;

    FOR1(i,n) rank[sa[i]]=i;

    FOR0(i,n)

    {

        if(k) k--;

        j=sa[rank[i]-1];

        while(i+k<n&&j+k<n&&r[i+k]==r[j+k]) k++;

        h[rank[i]]=k;

    }

}





int n;

char s1[N],s2[N];





int main()

{

    while(scanf("%s%s",s1,s2)!=-1)

    {

        n=strlen(s1);

        s1[n]='$';

        s1[n+1]=0;

        strcat(s1,s2);

        int i,M=strlen(s1);

        FOR0(i,M) r[i]=s1[i];

        r[M]=0;

        da(r,sa,M+1,130);

        calHeight(r,sa,M);

        int ans=0;

        FOR(i,2,M) if(h[i]>ans&&(i64)(sa[i]-n)*(sa[i-1]-n)<0)

        {

            ans=h[i];

        }

        PR(ans);

    }

    return 0;

}