POJ2774：Long Long Message——题解

最新推荐文章于 2020-07-04 11:57:59 发布

weixin_34129696

最新推荐文章于 2020-07-04 11:57:59 发布

阅读量115

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/luyouqi233/p/8466366.html

本文介绍了一种求解两个字符串最长公共子串的问题，并提供了一个详细的C++实现方案。通过合并字符串并利用后缀数组和高度数组来高效计算最长公共子串的长度。

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

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

论文题，把两个串合并起来，比较两个串各自的后缀的height值取最大即可。

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cctype>
#include<cstdio>
#include<vector>
#include<queue>
#include<cmath>
using namespace std;
const int N=200010;
char s[N];
int n,len,rank[N],sa[N],height[N],w[N];
inline bool pan(int *x,int i,int j,int k){
    int ti=i+k<n?x[i+k]:-1;
    int tj=j+k<n?x[j+k]:-1;
    return x[i]==x[j]&&ti==tj;
}
inline void SA_init(){
    int *x=rank,*y=height,r=256;
    for(int i=0;i<r;i++)w[i]=0;
    for(int i=0;i<n;i++)w[s[i]]++;
    for(int i=1;i<r;i++)w[i]+=w[i-1];
    for(int i=n-1;i>=0;i--)sa[--w[s[i]]]=i;
    r=1;x[sa[0]]=0;
    for(int i=1;i<n;i++)
    x[sa[i]]=s[sa[i]]==s[sa[i-1]]?r-1:r++;
    for(int k=1;r<n;k<<=1){
    int yn=0;
    for(int i=n-k;i<n;i++)y[yn++]=i;
    for(int i=0;i<n;i++)
        if(sa[i]>=k)y[yn++]=sa[i]-k;
    for(int i=0;i<r;i++)w[i]=0;
    for(int i=0;i<n;i++)++w[x[y[i]]];
    for(int i=1;i<r;i++)w[i]+=w[i-1];
    for(int i=n-1;i>=0;i--)sa[--w[x[y[i]]]]=y[i];
    swap(x,y);r=1;x[sa[0]]=0;
    for(int i=1;i<n;i++)
        x[sa[i]]=pan(y,sa[i],sa[i-1],k)?r-1:r++;
    }
    for(int i=0;i<n;i++)rank[i]=x[i];
}
inline void height_init(){
    int i,j,k=0;
    for(i=1;i<=n;i++)rank[sa[i]]=i;
    for(i=0;i<n;i++){
    if(k)k--;
    else k=0;
    j=sa[rank[i]-1];
    while(s[i+k]==s[j+k])k++;
    height[rank[i]]=k;
    }
}
int solve(){
    int ans=0;
    for(int i=2;i<n;i++){
    if(height[i]>ans){
        if(0<=sa[i-1]&&sa[i-1]<len&&len<sa[i]){
        ans=height[i];
        }
        if(0<=sa[i]&&sa[i]<len&&len<sa[i-1]){
        ans=height[i];
        }
    }
    }
    return ans;
}
int main(){
    scanf("%s",s);
    len=n=strlen(s);
    s[n++]=123;
    scanf("%s",s+n);
    n=strlen(s);
    s[n++]=0;
    SA_init();
    n--;
    height_init();
    printf("%d\n",solve());
    return 0;
}