[POJ 2774] 最长公共子串

最新推荐文章于 2016-09-01 17:16:20 发布

zhshrzhshr

最新推荐文章于 2016-09-01 17:16:20 发布

阅读量319

点赞数

分类专栏：后缀数组

本文链接：https://blog.csdn.net/zhshrzhshr/article/details/50358443

版权

后缀数组专栏收录该内容

2 篇文章 0 订阅

订阅专栏

题目大意：给定两个字符串，求它们的最长公共子串（废话）。

后缀数组大模板题。
两串接起来，height数组求个最大的且分别在两串中的就行了。
啥都不说贴代码。

#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <string>
#include <cstdlib>
#include <algorithm>
#include <vector>
#include <queue>
#include <deque>
#include <map>
#include <set>
#include <ctime>
#define putarray(a,n) for (int i=1;i<=n-1;i++) printf("%d ",a[i]); printf("%d\n",a[n]);
using namespace std;

const int MAXN=100005;
char A[MAXN],B[MAXN]; int ans,l1,l2; 

class Suffix_array
{   
    char str[MAXN<<1]; 
    int len,sa[MAXN<<1],rank[MAXN<<1],height[MAXN<<1],C[MAXN<<1];

    void build_sa(int m)
      {     
        int *x=rank,*y=height; //这里rank,height只是拿来缓存的。。。。 
        for (int i=0;i<=m;i++) C[i]=0;
        for (int i=1;i<=len;i++) x[i]=int(str[i]-96),C[x[i]]++;
        for (int i=1;i<=m;i++) C[i]+=C[i-1]; 
        for (int i=1;i<=len;i++) sa[C[x[i]]--]=i; 
        for (int k=1;k<=len;k<<=1)
          {
            int p=0; 
            for (int i=len-k+1;i<=len;i++) y[++p]=i;
            for (int i=1;i<=len;i++) if (sa[i]-k>0) y[++p]=sa[i]-k;
            for (int i=0;i<=m;i++) C[i]=0;
            for (int i=1;i<=len;i++) C[x[y[i]]]++;
            for (int i=1;i<=m;i++) C[i]+=C[i-1]; 
            for (int i=len;i>=1;i--) sa[C[x[y[i]]]--]=y[i];
            swap(x,y); p=1; x[sa[1]]=1;
            for (int i=2;i<=len;i++)
              if (y[sa[i]]==y[sa[i-1]] && y[sa[i]+k]==y[sa[i-1]+k]) x[sa[i]]=p;else x[sa[i]]=++p;

            if (p==len) break;
            m=p;
          }
      }

    void build_height()
      { 
        int k=0; 
        for (int i=1;i<=len;i++) rank[sa[i]]=i;
        for (int i=1;i<=len;i++)
          {
            if (k) k--; int j=sa[rank[i]-1];
            while (str[i+k]==str[j+k]) k++;
            height[rank[i]]=k;
          }
      }

    public: 

    void init()
      { 
        int _max=0; l1=strlen(A),l2=strlen(B); len=l1+l2+1; 
        for (int i=0;i<l1;i++) str[i+1]=A[i]; str[l1+1]=96;
        for (int i=0;i<l2;i++) str[i+l1+2]=B[i]; 
        build_sa(26);
        build_height();
      }

    void solve()
      { 
        ans=0;
        for (int i=1;i<=l1;i++) if (sa[rank[i]-1]>l1+1) ans=max(ans,height[rank[i]]);
        for (int i=l1+1;i<=len;i++) if (sa[rank[i]-1]<=l1) ans=max(ans,height[rank[i]]);
      }
}SA; 

int main()
{
    //freopen("Longest common substring.in","r",stdin);
    //freopen("Longest common substring.out","w",stdout);
    scanf("%s%s",A,B); 
    SA.init(); 
    SA.solve(); 
    printf("%d\n",ans);
    return 0;
}