URAL 1517. Freedom of choice(后缀数组:最长公共连续子串)

URAL 1517. Freedom of choice(后缀数组:最长公共连续子串)

题意:

        给你两个等长的串,求他们的最长公共连续子串.

分析: 

        罗穗骞《后缀数组——处理字符串的有力工具》例题.

        其实本题就是POJ2774的略微加强版:

http://blog.csdn.net/u013480600/article/details/24045165

        这里要求我们输出这个最长公共串了.(题目中要求有多解就输出所有解,但是其实只有1个解.)

        对于输出解,我们只要记录过程中的每一个可以作为解的后缀编号以及长度即可一一输出.

AC代码:

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=200000+1000;
int len1,len2;//分别为串1和串2的长度
int ans,pos;//ans是公共字串的长度,pos是起始位置
struct SuffixArray
{
    char s[maxn];
    int sa[maxn],rank[maxn],height[maxn];
    int t1[maxn],t2[maxn],c[maxn],n;
    void build_sa(int m)
    {
        int i,*x=t1,*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(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[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]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k]?p-1:p++;
            if(p>=n) break;
            m=p;
        }
    }
    void build_height()
    {
        int i,j,k=0;
        for(i=0;i<n;i++) rank[sa[i]]=i;
        for(i=0;i<n;i++)
        {
            if(k)k--;
            j=sa[rank[i]-1];
            while(s[i+k]==s[j+k]) k++;
            height[rank[i]]=k;
        }
    }
    void solve()
    {
        ans=0;//最长字串的长度
        for(int i=2;i<n;i++)
        {
            int a1=sa[i-1],a2=sa[i];
            if(a1>a2) swap(a1,a2);
            if(a1>=0&&a1<=len1-1&&a2>=len1+1&&a2<=len1+len2)
            {
                if(ans<height[i])
                {
                    ans = height[i];
                    pos=a1;
                }
            }

        }
    }
}sa;
int main()
{
    scanf("%d",&len1);
    len2=len1;
    scanf("%s",sa.s);
    sa.s[len1]=1;
    scanf("%s",sa.s+len1+1);
    sa.n=len1+len2+2;
    sa.s[len1+len2+1]=0;
    sa.build_sa(128);
    sa.build_height();
    sa.solve();
    sa.s[pos+ans]=0;
    printf("%s\n",sa.s+pos);
    return 0;
}