poj2774 Long Long Message（后缀数组）

 

【题目链接】

 

　　http://poj.org/problem?id=2774

 

【题意】

 

　　A & B的最长公共子序列。　　

 

【思路】

       拼接+height数组。将AB拼接成一个形如A$B的串，枚举height数组并判断sa[i]是否与sa[i-1]分别属于两个不同的字符串，如果是则比较得ans。

　　 时间复杂度为O(nlogn)。

       Ps：注意不能直接取height的最大值，因为取到的两个后缀的lcp可能同处于一个字符串。

【代码】

#include<cstdio>
#include<cstring>
#include<iostream>
#define FOR(a,b,c) for(int a=(b);a<=(c);a++)
using namespace std;

const int maxn = 300000+10; 
const int maxm = 150;

int s[maxn];
int sa[maxn],c[maxm],t[maxn],t2[maxn];

void build_sa(int m,int n) {
    int i,*x=t,*y=t2;
    for(i=0;i<m;i++) c[i]=0;
    for(i=0;i<n;i++) c[x[i]=s[i]]++;
    for(i=1;i<m;i++) c[i]+=c[i-1];
    for(i=n-1;i>=0;i--) sa[--c[x[i]]]=i;
    
    for(int k=1;k<=n;k<<=1) {
        int p=0;
        for(i=n-k;i<n;i++) y[p++]=i;
        for(i=0;i<n;i++) if(sa[i]>=k) y[p++]=sa[i]-k;
        
        for(i=0;i<m;i++) c[i]=0;
        for(i=0;i<n;i++) c[x[y[i]]]++;
        for(i=0;i<m;i++) c[i]+=c[i-1];
        for(i=n-1;i>=0;i--) sa[--c[x[y[i]]]]=y[i];
        
        swap(x,y);
        p=1; x[sa[0]]=0;
        for(i=1;i<n;i++) 
            x[sa[i]]=y[sa[i]]==y[sa[i-1]] && y[sa[i]+k]==y[sa[i-1]+k]?p-1:p++;
        if(p>=n) break;
        m=p;
    }
}
int rank[maxn],height[maxn];
void getHeight(int n) {
    int i,j,k=0;
    for(i=0;i<=n;i++) rank[sa[i]]=i;
    for(i=0;i<n;i++) {
        if(k) k--;
        j=sa[rank[i]-1];
        while(s[j+k]==s[i+k]) k++;
        height[rank[i]]=k;
    }
}

char a[maxn],b[maxn];
int n;

int main() {
    scanf("%s%s",a,b);
    int lena=strlen(a),lenb=strlen(b);
    n=lena+lenb+1;
    for(int i=0;i<lena;i++) s[i]=a[i]; s[lena]=1;
    for(int i=0;i<lenb;i++) s[lena+i+1]=b[i];
    s[n]=0;
    
    build_sa('z'+1,n+1);
    getHeight(n);
    
    int ans=0;
    for(int i=2;i<=n;i++) {
        int k=0;
        k+=sa[i]<lena?1:2;
        k+=sa[i-1]<lena?1:2;
        if(k==3) ans=max(ans,height[i]);
    }
    printf("%d
",ans);
    return 0;
}

UPD.16/4/6

当然也可以用SAM搞一搞，写起来更简单些