poj 1743 Musical Theme 【后缀数组】

有希望成为最长公共前缀不小于 k 的两个后缀一定在同一组。然后对于每组后缀，只须判断每个后缀的 sa 值的最大值和最小值之差是否不小于k。如果有一组满足，则说明存在，否则不存在。

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
const int maxn=20100;

int a[maxn],b[maxn],s[maxn],c[maxn];
int *rank,*height,*sa=s+1;
int num[maxn];
void sortandrank(int *a1,int *a2,int n,int &m,int j)
{
    int i;memset(c,0,sizeof(c));
    for(i=0;i< n;i++) c[a1[i]]++;
    for(i=1;i<=m;i++) c[i]+=c[i-1];
    for(i=n-1;i>=0;i--) sa[--c[a1[a2[i]]]]=a2[i];
    a2[sa[0]]=m=0;
    for(i=1;i<n;i++) a2[sa[i]]=a1[sa[i-1]]==a1[sa[i]]&&a1[sa[i-1]+j]==a1[sa[i]+j]?m:++m;
}

void da(int*str,int n,int m)
{
    int *a1=a,*a2=b,*tmp;
    int i,j,p;
    for(i=0;i<n;i++) a1[i]=i,a2[i]=str[i];
    a1[n]=a2[n]=-1;
    sortandrank(a2,a1,n,m,0);
    for(j=1;m<n-1;j<<=1)
    {
        p=0;
        for(i=n-j;i<n;i++) a2[p++]=i;
        for(i=0;i<n;i++) if(sa[i]>=j) a2[p++]=sa[i]-j;
        sortandrank(a1,a2,n,m,j);
        tmp=a1,a1=a2,a2=tmp;
    }
    rank=a1,height=a2;
}
void calheight(int *str,int n)
{
    int i,j,k;
    sa[-1]=n;
    for(height[0]=k=i=0;i<n;i++)
    {
        for(k?k--:0,j=sa[rank[i]-1];str[i+k]==str[j+k];k++);
        height[rank[i]]=k;
    }
}
inline int abs(int a){return a>0?a:-a;}
bool jud(int mid,int n){
    for(int i=2;i<=n;i++){
        if(height[i]<mid)continue;
        for(int j=i-1;j>=2&&height[j+1]>=mid;j--)
            if(abs(sa[i]-sa[j])>=mid) return 1;
    }
    return 0;
}
int main()
{
    int n;
    while(scanf("%d",&n),n)
    {
        int x,y;
        for(int i=0;i<n;i++) scanf("%d",num+i);
        for(int i=0;i<n-1;i++) num[i]=num[i+1]-num[i]+90;
        num[n-1]=0;
        if(n<10){puts("0");continue;}
        da(num,n,180);calheight(num,n-1);
        int l=3,r=n/2+1,mid,ans;
        while(l<r)
        {
            mid=(l+r)>>1;
            if(jud(mid,n)) l=mid+1,ans=mid;
            else r=mid;
        }
        printf("%d\n",ans<4?0:ans+1);
    }
    return 0;
}