Secretary POJ - 2217 （最长公共子串问题&&后缀数组与高度数组的运用）

题目链接

vjudge题目链接

题意

给定两个串a，b。计算两个字符串的最长公共子串的长度。

分析

先考虑简化问题：
求在一个串中至少出现两次的最长子串。
答案就是在后缀数组中相邻的后缀的最长公共前缀。因为在后缀数组中的起始位置相距越远，他们的最长公共前缀就越小。所以只要求出高度数组的最大值即可。

问题转化：
将两个字符串连接起来并在连接处添加一个字符’$’，形成新串s.这样求在s中至少出现两次的合适的最长子串就是a、b的最长公共子串。合适的要求是两个子串的起始位置要一个在a中，一个在b中。

79ms的代码

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

const int maxn=2e4+100;
int n,k,rank[maxn],tmp[maxn],sa[maxn],lcp[maxn];
char a[maxn],b[maxn],s[maxn];

bool cmp_sa(int i,int j)//用倍增法对sa排序
{
    if(rank[i]!=rank[j]) return rank[i]<rank[j];
    else
    {
        int ri=i+k<=n?rank[i+k]:-1;
        int rj=j+k<=n?rank[j+k]:-1;
        return ri<rj;
    }
}
void get_sa()//求后缀数组
{
    for(int i=0;i<=n;i++)
    {
        sa[i]=i;
        rank[i]=i<n?s[i]:-1;
    }
    for(k=1;k<=n;k*=2)
    {
        sort(sa,sa+n+1,cmp_sa);
        tmp[sa[0]]=0;
        for(int i=1;i<=n;i++)
        {
            tmp[sa[i]]=tmp[sa[i-1]]+(cmp_sa(sa[i-1],sa[i])?1:0);

        }
        for(int i=0;i<=n;i++) rank[i]=tmp[i];
    }
}
void get_lcp()//求高度数组
{
    for(int i=0;i<=n;i++) rank[sa[i]]=i;
    int h=0;
    lcp[0]=0;
    for(int i=0;i<n;i++)
    {
        int j=sa[rank[i]-1];
        if(h>0) h--;
        for(;i+h<n && j+h<n;h++)
        {
            if(s[i+h]!=s[j+h]) break;
        }
        lcp[rank[i]-1]=h;
    }
}
int main()
{
    int N;
    scanf("%d",&N);
    getchar();
    while(N--)
    {
        //输入
        gets(a);
        gets(b);
        n=0;
        int len1=strlen(a);
        int len2=strlen(b);
        //形成新串s
        for(int i=0;i<len1;i++)
            s[n++]=a[i];
        s[n++]='$';
        for(int i=0;i<len2;i++)
            s[n++]=b[i];
        s[n]='\0';
        //计算s的后缀数组和高度数组
        get_sa();
        get_lcp();
        //找合适的最大的高度数组值
        int ans=0;
        for(int i=0;i<n;i++)
        {
            if((sa[i]<len1) != (sa[i+1]<len1))
                ans=max(ans,lcp[i]);
        }
        printf("Nejdelsi spolecny retezec ma delku %d.\n",ans);
    }
    return 0;
}

模板链接

高度数组
后缀数组