POJ 1743 后缀数组

题目大意

找到一个连续的子序列连续出现过两次，且这两次不相交，只要子序列中每个数都加/减一个数得到一个新的序列，也可以看作是相同的

 

那么也就是说这道题目可以转化成找到两个子序列，这两个子序列中每个数前后的差值是相等的

所以我们可以求所有数两两之间的差值，然后根据这个值建立后缀数组后，二分答案来解决问题

 

#include <cstdio>
#include <cstring>
#include <iostream>

using namespace std;
const int N = 20010;
int rank[N] , sa[N] , height[N];
int wa[N] , wb[N] , tmp[N] , wv[N];
int a[N] , b[N];

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

void getSa(int *r , int *sa , int n , int m)
{
    int i,j,p;
    int *x=wa , *y=wb , *t;
    for(i=0 ; i<m ; i++) tmp[i]=0;
    for(i=0 ; i<n ; i++) tmp[x[i]=r[i]]++;
    for(i=1 ; i<m ; i++) tmp[i]+=tmp[i-1];
    for(i=n-1 ; i>=0 ; i--) sa[--tmp[x[i]]]=i;

    //初始p=1是防止只有一个元素就可以退出循环的
    for(p=1,j=1 ; p<n ; j*=2,m=p){
        //对第二关键字排序
        p=0;
        for(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<n ; i++) wv[i]=x[y[i]];
        for(i=0 ; i<m ; i++) tmp[i]=0;
        for(i=0 ; i<n ; i++) tmp[wv[i]]++;
        for(i=1 ; i<m ; i++) tmp[i]+=tmp[i-1];
        for(i=n-1 ; i>=0 ; i--) sa[--tmp[wv[i]]]=y[i];
        t=x,x=y,y=t;
        x[sa[0]]=0;
        for(p=1,i=1 ; i<n ; i++)
            x[sa[i]] = cmp(y,sa[i-1],sa[i],j)?p-1:p++;
    }

    return;
}

void callHeight(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++);
    return;
}

bool check(int m , int n)
{
    int l=sa[0] , r=sa[0];
    for(int i=1 ; i<n ; i++){
        if(height[i]<m){
            l=sa[i] , r=sa[i];
        }
        else{
            l=min(sa[i] , l);
            r=max(sa[i] , r);
            if(r-l>m) return true;
        }
    }
    return false;
}

int main()
{
  //  freopen("a.in" , "r" , stdin);
    int n;
    while(scanf("%d" , &n) , n)
    {
        for(int i=0 ; i<n ; i++) scanf("%d" , &a[i]);
        for(int i=0 ; i<n-1 ; i++) b[i]=a[i+1]-a[i]+90;
        b[n-1]=0;
        getSa(b,sa,n,200);
        callHeight(b,sa,n-1);

        int l=0 , r=n , ans=0;
        while(l<=r){
            int m=(l+r)>>1;
            if(!check(m,n))  r=m-1;
            else{
                ans=m;
                l=m+1;
            }
        }
        if(ans>=4) printf("%d\n" , ans+1);
        else puts("0");
    }
    return 0;
}