POJ - 2774Long Long Message (神tm的后缀数组)

两个字符串的最长公共子串。

sa 排名是i的子串的位置在哪。

rak 位置是i 的子串排名是多少。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn=200000+100;
char s[maxn],s2[maxn];
int sa[maxn],t[maxn],t2[maxn],c[maxn];
int height[maxn],rk[maxn];
void build_sa(int n,int m)
{
    int i,*x=t,*y=t2;
    for(i=0;i<m;++i) c[i]=0;
    for(i=0;i<n;++i) x[i]=s[i];
    for(i=0;i<n;++i) c[x[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[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) rk[sa[i]]=i;
    for(i=0;i<n;++i)
    {
        if(k) k--;
        j=sa[rk[i]-1];
        while(s[i+k]==s[j+k]) k++;
        height[rk[i]]=k;
    }
}
int main()
{
    int n1,n2,n;
    scanf("%s",s);
    scanf("%s",s2);
    n1=strlen(s);
    n2=strlen(s2);
    n=n1+n2+1;
    s[n1]='$';
    for(int i=0;i<n2;++i)
        s[n1+i+1]=s2[i];
    build_sa(n,150);
    getHeight(n);
    int maxlen=-1;
    for(int i=1;i<n;++i)
        if((sa[i] < n1) != (sa[i-1] < n1))
            maxlen = max(height[i],maxlen);
    printf("%d\n",maxlen);
    return 0;
}
 

猜你喜欢

转载自blog.csdn.net/kidsummer/article/details/81782506