[Poj1743] [后缀数组论文例题] Musical Theme [后缀数组不可重叠最长重复子串]

利用后缀数组，先对读入整数处理str[i]=str[i+1]-str[i]+90这样可以避免负数，计算Height数组，二分答案，如果某处H<lim则将H数组分开，最终分成若干块，判断每块中是否存在SA[i]-SA[j]>lim（注意不是>=因为查分后如果相等那么两个公共串连接部分的元素是公用的，不符合题意），注意特判n=1的情况，因为查分后的结果是个空串，会导致RE。

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

using namespace std;

int    A[21000],B[21000],U[21000],SA[21000],H[21000];
int    str[21000];

void    Calc_H(const int n,const int * Rank)
{
    int    i,j,k=0;
    for(i=0;i<n;H[Rank[i++]]=k)
        for(k?k--:0,j=SA[Rank[i]-1];str[i+k]==str[j+k];++k);
    return ;
}

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

int*    Get_SA(const int n,int m)
{
    int    i,j,p,*x=A,*y=B;
    memset(U,0,sizeof(U));
    for(i=0;i<n;++i)U[x[i]=str[i]]++;
    for(i=1;i<m;++i)U[i]+=U[i-1];
    for(i=n-1;i>=0;--i)SA[--U[x[i]]]=i;

    for(j=1,p=1;p<n;j<<=1,m=p)
    {
        for(p=0,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)U[i]=0;
        for(i=0;i<n;++i)U[x[y[i]]]++;
        for(i=1;i<m;++i)U[i]+=U[i-1];
        for(i=n-1;i>=0;--i)SA[--U[x[y[i]]]]=y[i];
        for(swap(x,y),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++;
    }
    Calc_H(n,x);
    return x;
}

bool    Check(const int lim,const int n)
{
    int    Min=SA[1],Max=SA[1];
    for(int i=1;i<=n;++i)
    {
        if(i>1 && H[i]<lim)
        {
            if(Max-Min>=lim+1)return true;
            Min=0x7fffffff;
            Max=-1;
        }
        if(SA[i]<Min)Min=SA[i];
        if(SA[i]>Max)Max=SA[i];
    }
    if(Max-Min>lim)return true;
    return false;
}

void    Init()
{
    memset(A,0,sizeof(A));
    memset(B,0,sizeof(B));
    memset(U,0,sizeof(U));
    memset(H,0,sizeof(H));
    memset(SA,0,sizeof(SA));
    return ;
}

int main()
{
    int    n,i,l,r;

    while(~scanf("%d",&n) && n)
    {
        Init();
        for(i=0;i<n;++i)scanf("%d",&str[i]);
        if(n==1){printf("0
");continue;}
        n--;
        for(i=0;i<n;++i)str[i]=str[i+1]-str[i]+90;
        str[n]=0;

        Get_SA(n+1,200);

        l=0,r=n;
        while(l<r-1)
        {
            int    mid=l+((r-l)>>1);
            if(Check(mid,n))l=mid;
            else        r=mid;
        }
        printf("%d
",l>=4 ? l+1:0);
    }

    return 0;
}