Codevs 3160 最长公共子串(后缀数组)

3160 最长公共子串
时间限制: 2 s
空间限制: 128000 KB
题目等级 : 大师 Master
题目描述 Description
给出两个由小写字母组成的字符串，求它们的最长公共子串的长度。
输入描述 Input Description
读入两个字符串
输出描述 Output Description
输出最长公共子串的长度
样例输入 Sample Input
yeshowmuchiloveyoumydearmotherreallyicannotbelieveit
yeaphowmuchiloveyoumydearmother
样例输出 Sample Output
27
数据范围及提示 Data Size & Hint
单个字符串的长度不超过100000
分类标签 Tags
后缀树 后缀树组 树结构

/*
后缀数组求lcp裸题.
先用无关字符将两个串连接起来.
然后用后缀数组求一下rank.
这样答案显然就是两个相邻rank后缀的最大前缀.
这样的话.
求出height数组,
然后找一个i和i-1分别属于前后两个串就可以了.
*/
#include<iostream>
#include<cstring>
#include<cstdio>
#define MAXN 2000010
using namespace std;
int n,sa[MAXN],l1,l2,m=128,rank[MAXN],height[MAXN],t1[MAXN],c[MAXN],t2[MAXN],ans,tot,s[MAXN];
char ch1[MAXN],ch2[MAXN],ch[MAXN];
bool cmp(int *y,int a,int b,int k)
{
    int a1=y[a],b1=y[b];
    int a2=a+k>=n?-1:y[a+k];
    int b2=b+k>=n?-1:y[b+k];
    return a1==b1&&a2==b2;
}
void slovesa()
{
    int *x=t1,*y=t2;
    for(int i=0;i<m;i++) c[i]=0;
    for(int i=0;i<n;i++) c[x[i]=s[i]]++;
    for(int i=1;i<m;i++) c[i]+=c[i-1];
    for(int i=n-1;i>=0;i--) sa[--c[x[i]]]=i;
    for(int k=1,p=0;k<=n;k<<=1,m=p,p=0)
    {
        for(int i=n-k;i<n;i++) y[p++]=i;
        for(int i=0;i<n;i++) if(sa[i]>=k) y[p++]=sa[i]-k;
        for(int i=0;i<m;i++) c[i]=0;
        for(int i=0;i<n;i++) c[x[y[i]]]++;
        for(int i=1;i<m;i++) c[i]+=c[i-1];
        for(int i=n-1;i>=0;i--) sa[--c[x[y[i]]]]=y[i];
        swap(x,y);p=1;x[sa[0]]=0;
        for(int i=1;i<n;i++)
        {
            if(cmp(y,sa[i-1],sa[i],k)) x[sa[i]]=p-1;
            else x[sa[i]]=p++;
        }
        if(p>=n) break;
    }
}
void sloveheight()
{
    int k=0;
    for(int i=0;i<n;i++) rank[sa[i]]=i;
    for(int i=0;i<n;height[rank[i++]]=k)
    {
        if(!rank[i]) continue;
        int j=sa[rank[i]-1];
        if(k) k--;
        while(j+k<n&&i+k<n&&s[j+k]==s[i+k]) k++;
    }
    height[0]=0;
}
void sloveans()
{
    for(int i=0;i<n;i++)
      if(height[i]>ans)
      {
        if(sa[i]<l1&&sa[i-1]>l1) ans=height[i];
        if(sa[i]>l1&&sa[i-1]<l1) ans=height[i];
      }
}
int main()
{
    scanf("%s",ch1);
    scanf("%s",ch2);
    l1=strlen(ch1);
    l2=strlen(ch2);
    for(int i=0;i<l1;i++) s[n++]=ch1[i];
    s[n++]='$';
    for(int i=0;i<l2;i++) s[n++]=ch2[i];
    slovesa(),sloveheight(),sloveans();
    printf("%d",ans);
    return 0;
}