[poj1743]Musical Theme后缀数组

 1 #include <cstdlib>
 2 #include <cstring>
 3 #include <cstdio>
 4 #include <algorithm>
 5 #include<iostream>
 6 #include<cstdlib>
 7 #define inf 0x3f3f3f3f
 8 using namespace std;
 9 const int N=200005;
10 int wa[N],wb[N],wv[N],wc[N];
11 
12 bool cmp(int *r,int a,int b,int l){
13     return r[a]==r[b]&&r[a+l]==r[b+l];
14 }
15 
16 void make_sa(int *r,int *sa,int n,int m){//m是至多有多少个字符，up的意思
17     int i,j,p,*x=wa,*y=wb;
18     for(i=0;i<m;i++) wc[i]=0;
19     for(i=0;i<n;i++) wc[x[i]=r[i]]++;
20     for(i=1;i<m;i++) wc[i]+=wc[i-1];
21     for(i=n-1;i>=0;i--) sa[--wc[x[i]]]=i;
22     for(j=1,p=1;p<n;j*=2,m=p){
23         for(p=0,i=n-j;i<n;i++) y[p++]=i;
24         for(i=0;i<n;i++) if(sa[i]>=j) y[p++]=sa[i]-j;
25         for(i=0;i<n;i++) wv[i]=x[y[i]];
26         for(i=0;i<m;i++) wc[i]=0;
27         for(i=0;i<n;i++) wc[wv[i]]++;
28         for(i=1;i<m;i++) wc[i]+=wc[i-1];
29         for(i=n-1;i>=0;i--) sa[--wc[wv[i]]]=y[i];
30         for(swap(x,y),p=1,x[sa[0]]=0,i=1;i<n;i++)
31             x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;
32     }
33     return;
34 }
35 
36 int rank1[N],height[N],sa[N];//输出都是从1开始存的，rank从0开始，最后需要+1，读入从下标0
37 void make_height(int *r,int *sa,int n){
38     int i,j,k=0;
39     for(i=1;i<=n;i++) rank1[sa[i]]=i;
40     for(i=0;i<n;height[rank1[i++]]=k)
41         for(k?k--:0,j=sa[rank1[i]-1];r[i+k]==r[j+k];k++);
42     return;
43 }
44 
45 int r[N],a[N];
46 int n,k;
47 bool check(int k){
48     int maxn,minn;
49     maxn=-inf,minn=inf;
50     for(int i=2;i<=n;i++){
51         if(height[i]>=k){
52             minn=min(min(sa[i],sa[i-1]),minn);//lcp一定在height中
53             maxn=max(max(sa[i],sa[i-1]),maxn);
54             if(maxn-minn>k) return 1;//这是距离
55         }else        maxn=-inf,minn=inf;
56 
57     }
58     return 0;
59 }
60 int erfen(int l,int r){
61     while(l<r){
62         int mid=(l+r+1)>>1;
63         if(check(mid)) l=mid;
64         else r=mid-1;
65     }
66     l++;
67     if(l<5) l=0;
68     return l;
69 }
70 
71 int main(){
72     ios::sync_with_stdio(0);
73     cin.tie(0);
74     cout.tie(0);
75     while(cin>>n&&n){
76         for(int i=0;i<n;i++) cin>>a[i];
77         for(int i=0;i<n-1;i++) r[i]=a[i+1]-a[i]+100;
78         r[--n]=0;
79         make_sa(r, sa, n+1,220);
80         make_height(r, sa, n);
81         int ans=erfen(0,n);
82         cout<<ans<<"\n";
83     }
84     return 0;
85 }