最长重复子串（可重叠） 后缀数组

http://www.bianchengla.com/course/ds/practise/problem?id=1387

找了半天终于找到一个可以提交的地方。。。

 

题解：

任何一个重复子串，都必然是某两个后缀的最长公共前缀。

因为，两个后缀的公共前缀，它出现在这两个后缀中，并且起始位置时不同的，所以这个公共前缀必然重复出现两次以上（可重叠）。

而任何两个后缀的最长公共前缀为某一段height值中的最小值，所以最大为height值中的最大值（即某个lcp(sa[i],sa[i+1]））。

所以只要算出height数组，然后输出最大值就可以了。

 

#include <iostream>
#include <cstring>
#include <cstdlib>
#include <cstdio>
#include <algorithm>

#define N 50050

using namespace std;

int wa[N],wb[N],wc[N],wv[N];
int r[N],sa[N];
char str[N];
int rank[N],height[N];

inline bool cmp(int *r,int a,int b,int l)
{
    return r[a]==r[b]&&r[a+l]==r[b+l];
}

inline void da(int *r,int *sa,int n,int m)
{
    int i,j,p,*x=wa,*y=wb,*t;
    for(i=0;i<m;i++) wc[i]=0;
    for(i=0;i<n;i++) wc[x[i]=r[i]]++;
    for(i=1;i<m;i++) wc[i]+=wc[i-1];
    for(i=n-1;i>=0;i--) sa[--wc[x[i]]]=i;
    for(j=1,p=1;p<n;j<<=1,m=p)
    {
        for(i=n-j,p=0;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<n;i++) wv[i]=x[y[i]];
        for(i=0;i<m;i++) wc[i]=0;
        for(i=0;i<n;i++) wc[wv[i]]++;
        for(i=1;i<m;i++) wc[i]+=wc[i-1];
        for(i=n-1;i>=0;i--) sa[--wc[wv[i]]]=y[i];
        for(t=x,x=y,y=t,p=1,x[sa[0]]=0,i=1;i<n;i++)
            x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;
    }
}

inline void getheight(int *r,int *sa,int n)
{
    int i,j,k=0;
    for(i=1;i<=n;i++) rank[sa[i]]=i;
    for(i=0;i<n;height[rank[i++]]=k)
        for(k?k--:0,j=sa[rank[i]-1];r[i+k]==r[j+k];k++);
}

inline void go()
{
    scanf("%s",str);
    int n=strlen(str);
    for(int i=0;i<n;i++) r[i]=str[i];
    r[n]=0;
    da(r,sa,n+1,256);
    getheight(r,sa,n);
    
    int ans=-1,pos;
    for(int i=1;i<=n;i++)
        if(height[i]>ans) ans=height[i],pos=sa[i];
    for(int i=pos;i<pos+ans;i++) printf("%c",str[i]);
    puts("");
}

int main()
{
    int cas;scanf("%d",&cas);
    while(cas--) go();
    return 0;
}