POJ2774-Long Long Message-后缀数组

后缀数组处理最长公共子串问题。

将两个串接在一起，计算height的最大值。同时要保证sa[i]和sa[i-1]在不同的串当中。

#include <cstdio>
#include <algorithm>
#include <cstring>

using namespace std;

const int maxn = 2e5+10;
char s[maxn];
int sa[maxn],t[maxn],t2[maxn],c[maxn];
int rank[maxn],height[maxn];

void build(int n,int m)
{
    int i,j,p,*x = t, *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(j=1;j<=n;j<<=1)
    {
        p=0;
        for(i=n-j;i<n;i++)y[p++]=i;
        for(i=0;i<n;i++)if(sa[i]>=j)y[p++]=sa[i]-j;
        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-1]]==y[sa[i]]&&y[sa[i-1]+j]==y[sa[i]+j]?p-1:p++;
        if(p>=n)break;
        m=p;
    }
    n--;
    int k = 0;
    for(int i=0;i<=n;i++) rank[sa[i] ] = i;
    for(int i=0;i<n;i++)
    {
        if(k) k--;
        int j = sa[rank[i]-1];
        while(s[i+k]==s[j+k]) k++;
        height[rank[i] ] = k;
    }
}

char line_1[maxn],line_2[maxn];

bool check(int a,int b,int len)
{
    if(a > b) swap(a,b);
    if(a < len && b > len)
    {
        return true;
    }
    return false;
}

int main()
{
    while(~scanf("%s%s",line_1,line_2))
    {
        int len1 = strlen(line_1);
        int len2 = strlen(line_2);
        int n = len1+len2+1;
        for(int i=0;i<len1;i++) s[i] = line_1[i];
        s[len1] = 1;
        for(int i=0;i<len2;i++) s[len1+1+i] = line_2[i];
        s[len1+len2+1] = 0;

        //printf("%s
",s);
        build(n+1,128);
        int ans = 0;
        for(int i=1;i<n;i++)
        {
            if(check(sa[i],sa[i-1],len1))
                ans = max(ans,height[i]);
        }
        printf("%d
",ans);
    }
}