POJ 1743 后缀数组

点击打开链接

题意：问最长的两段连续子串的长度，且两段不重复，这两段可以是这样的第一段的每个数字加或减同一个数，形成的第二段

思路：楼爷的男人8题果然有水准，用自己风格写的代码WA数十次，终于找到坑点，先将数字的差值计算出来，求后缀数组和高度数组，二分长度求出最长的，二分成功的条件为若二分值为mid，则对于有相同的公共前缀的长度大于mid的所有串中，最远距离减去最近距离的差值大于等于mid即可，具体看代码

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int MAXN=50010;
int wa[MAXN],wb[MAXN],wv[MAXN],ww[MAXN];
int sa[MAXN],lcp[MAXN],Rank[MAXN],rank1[MAXN];
inline bool cmp(int *r,int a,int b,int len){
    return r[a]==r[b]&&r[a+len]==r[b+len];
}
void construct_sa(int *str,int n,int m){
     int i,j,p,*x=wa,*y=wb,*t;
     for(i=0;i<m;i++) ww[i]=0;
     for(i=0;i<n;i++) ww[x[i]=str[i]]++;
     for(i=1;i<m;i++) ww[i]+=ww[i-1];
     for(i=n-1;i>=0;i--) sa[--ww[x[i]]]=i;
     for(j=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++) ww[i]=0;
         for(i=0;i<n;i++) ww[wv[i]=x[y[i]]]++;
         for(i=1;i<m;i++) ww[i]+=ww[i-1];
         for(i=n-1;i>=0;i--) sa[--ww[wv[i]]]=y[i];
         for(t=x,x=y,y=t,x[sa[0]]=0,p=i=1;i<n;i++)
             x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;
     }
}
void construct_lcp(int n,int *str){
    for(int i=0;i<=n;i++) rank1[sa[i]]=i;
    int h=0;
    lcp[0]=0;
    for(int i=0;i<n;i++){
        int j=sa[rank1[i]-1];
        if(h>0) h--;
        for(;j+h<n&&i+h<n;h++) if(str[i+h]!=str[j+h]) break;
        lcp[rank1[i]-1]=h;
    }
}
int num[MAXN];
bool judge(int len,int n){
    for(int i=0;i<n;i++){
        if(lcp[i]<len) continue;
        for(int j=i+1;j<=n;j++){
            if(abs(sa[j]-sa[i])>=len) return 1;
            if(lcp[j]<len) break;
        }
    }
    return 0;
}
int main(){
    int n;
    while(scanf("%d",&n)!=-1){
        if(n==0) break;
        for(int i=0;i<n;i++) scanf("%d",&num[i]);
        for(int i=1;i<n;i++) num[i-1]=num[i]-num[i-1]+90;
        construct_sa(num,n,300);
        construct_lcp(n-1,num);
        int le=0,ri=20010;
        while(ri-le>1){
            int mid=(le+ri)>>1;
            if(judge(mid,n-1)) le=mid;
            else ri=mid;
        }
        if(le<4) printf("0\n");
        else printf("%d\n",le+1);
    }
    return 0;
}