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;}