Long Long Message POJ

Long Long Message

POJ - 2774

题意：求两个串的最长公共字串。

用特殊符号连接两个字符串，后缀数组。

枚举height，如果sa[i]和sa[i-1]分别属于不同的串，则更新最大值。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn = 200010;
char s[maxn];
int sa[maxn],t1[maxn],t2[maxn],c[maxn];
int rank[maxn],height[maxn];


void build_sa(int n, int m){
    int i, *x = t1, *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 = 1; 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;
    }
}
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[i+k]==s[j+k]) k++;
        height[rank[i]] = k;
    }
}

int main(){
    while(scanf("%s",s)!=EOF){
        int n=strlen(s);
        int l1=n;
        s[n]='#';
        scanf("%s",s+n+1);
        n=strlen(s);
    //    cout<<s<<endl;
        build_sa(n+1,256);
        getHeight(n);
        int ans=0;
        for(int i = 1; i < n; i++){
            if(sa[i]<l1&&sa[i-1]>l1 || sa[i]>l1&&sa[i-1]<l1){
                ans=max(ans,height[i]);
            }
        }
        printf("%d
",ans);
    }
    return 0;
}