POJ2774 Long Long Message（后缀数组）

求两个字符串的最长公共子串。

拼接两个字符串中间用特殊字符隔开，max(height[i])（2<=i<=len，suffix(sa[i])与suffix(sa[i-1])分别属于两个字符串的后缀）就是答案！

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
#define MAXN 222222 
int wa[MAXN],wb[MAXN],wv[MAXN],ws[MAXN];
int cmp(int *r,int a,int b,int l){
    return r[a]==r[b] && r[a+l]==r[b+l];
}
int sa[MAXN],rank[MAXN],height[MAXN];
void SA(int *r,int n,int m){
    int *x=wa,*y=wb;

    for(int i=0; i<m; ++i) ws[i]=0;
    for(int i=0; i<n; ++i) ++ws[x[i]=r[i]];
    for(int i=1; i<m; ++i) ws[i]+=ws[i-1];
    for(int i=n-1; i>=0; --i) sa[--ws[x[i]]]=i;

    int p=1;
    for(int j=1; p<n; j<<=1,m=p){
        p=0;
        for(int i=n-j; i<n; ++i) y[p++]=i;
        for(int i=0; i<n; ++i) if(sa[i]>=j) y[p++]=sa[i]-j;
        for(int i=0; i<n; ++i) wv[i]=x[y[i]];
        for(int i=0; i<m; ++i) ws[i]=0;
        for(int i=0; i<n; ++i) ++ws[wv[i]];
        for(int i=1; i<m; ++i) ws[i]+=ws[i-1];
        for(int i=n-1; i>=0; --i) sa[--ws[wv[i]]]=y[i];
        swap(x,y); x[sa[0]]=0; p=1;
        for(int i=1; i<n; ++i) x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;
    }

    for(int i=1; i<n; ++i) rank[sa[i]]=i;
    int k=0;
    for(int i=0; i<n-1; height[rank[i++]]=k){
        if(k) --k;
        for(int j=sa[rank[i]-1]; r[i+k]==r[j+k]; ++k);
    }
}

int r[MAXN];
char str[MAXN];
int main(){
    int n=0,m;
    scanf("%s",str);
    for(int i=0; str[i]; ++i) r[n++]=str[i]-'a'+1;
    m=n;
    r[n++]=27;
    scanf("%s",str);
    for(int i=0; str[i]; ++i) r[n++]=str[i]-'a'+1;
    r[n]=0;
    SA(r,n+1,28);
    int res=0;
    for(int i=2; i<=n; ++i){
        if(sa[i]>m&&sa[i-1]<m || sa[i]<m&&sa[i-1]>m) res=max(res,height[i]);
    }
    printf("%d",res);
    return 0;
}