poj 3261 Milk Patterns 后缀数组+二分
2016-07-19 20:14
260 查看
/*********************************************************** 题目: Milk Patterns(poj 3261) 链接: http://poj.org/problem?id=3261 题意: 给一串数字,求这些数字中公共子串个数大于k的 最长串。 算法: 后缀数组+二分 ***********************************************************/ #include<iostream> #include<cstdio> #include<cstring> #include<algorithm> using namespace std; const int mx=20010; int s[mx],sa[mx],t[mx],t2[mx],c[mx*50],n,k; int rank[mx],height[mx]; void build_sa(int m) { int i,*x=t,*y=t2; for (i=0;i<m;i++) c[i]=0; for (i=0;i<n;i++) c[x[i]=s[i]]++; for (i=1;i<m;i++) c[i]+=c[i-1]; for (i=n-1;i>=0;i--) sa[--c[x[i]]]=i; for (int k=1;k<=n;k<<=1) { int p=0; for (i=n-k;i<n;i++) y[p++]=i; for (i=0;i<n;i++) if (sa[i]>=k) y[p++]=sa[i]-k; for (i=0;i<m;i++) c[i]=0; for (i=0;i<n;i++) c[x[y[i]]]++; for (i=1;i<m;i++) c[i]+=c[i-1]; for (i=n-1;i>=0;i--) sa[--c[x[y[i]]]]=y[i]; swap(x,y); p=1; x[sa[0]]=0; for (i=1;i<n;i++) x[sa[i]]=y[sa[i-1]]==y[sa[i]]&&y[sa[i-1]+k]==y[sa[i]+k]?p-1:p++; if (p>=n) break; m=p; } } void getHeight() { int i,j,k=0; for (i=0;i<n;i++) rank[sa[i]]=i; for (i=0;i<n-1;i++) { if (k) k--; int j=sa[rank[i]-1]; while (s[i+k]==s[j+k]) { k++; } height[rank[i]]=k; } } bool check(int len) { int i=0; while (i<n) { while (i<n&&height[i]<len) i++; if (i>=n) return false; int cut=1; while (height[i]>=len) { cut++; i++; } if (cut>=k) return true; } return false; } int getans() { int l=1,r=n-1; int ans=1; while (l<=r) { int m=(l+r)/2; if (check(m)) { l=m+1; ans=m; } else r=m-1; } return ans; } int main() { scanf("%d%d",&n,&k); for (int i=0;i<n;i++) { scanf("%d",&s[i]); } s =0; n++; build_sa(1000001); getHeight(); int ans=getans(); printf("%d\n",ans); }