1. 程式人生 > >POJ3261:Milk Patterns——題解

POJ3261:Milk Patterns——題解

題解 att block -s cto mil 重復子串 i++ clu

http://poj.org/problem?id=3261

給一個序列,求至少出現 k 次的最長重復子串,這 k 個子串可以重疊。

論文題+傻逼題。

上一道題(POJ1743)會做即可。

還是二分長度,按照高度數組分組做,統計每個組有多少個後綴即可。

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cctype>
#include<cstdio>
#include<vector>
#include<queue>
#include
<cmath> using namespace std; const int N=2e4+5; int n,m,rank[N],sa[N],height[N],w[N],s[N]; inline bool pan(int *x,int i,int j,int k){ int ti=i+k<n?x[i+k]:-1; int tj=j+k<n?x[j+k]:-1; return x[i]==x[j]&&ti==tj; } inline void SA_init(){ int *x=rank,*y=height,r=10000
; for(int i=0;i<r;i++)w[i]=0; for(int i=0;i<n;i++)w[s[i]]++; for(int i=1;i<r;i++)w[i]+=w[i-1]; for(int i=n-1;i>=0;i--)sa[--w[s[i]]]=i; r=1;x[sa[0]]=0; for(int i=1;i<n;i++) x[sa[i]]=s[sa[i]]==s[sa[i-1]]?r-1:r++; for(int k=1;r<n;k<<=1){ int
yn=0; for(int i=n-k;i<n;i++)y[yn++]=i; for(int i=0;i<n;i++) if(sa[i]>=k)y[yn++]=sa[i]-k; for(int i=0;i<r;i++)w[i]=0; for(int i=0;i<n;i++)++w[x[y[i]]]; for(int i=1;i<r;i++)w[i]+=w[i-1]; for(int i=n-1;i>=0;i--)sa[--w[x[y[i]]]]=y[i]; swap(x,y);r=1;x[sa[0]]=0; for(int i=1;i<n;i++) x[sa[i]]=pan(y,sa[i],sa[i-1],k)?r-1:r++; } for(int i=0;i<n;i++)rank[i]=x[i]; } inline void height_init(){ int i,j,k=0; for(i=1;i<=n;i++)rank[sa[i]]=i; for(i=0;i<n;i++){ if(k)k--; else k=0; j=sa[rank[i]-1]; while(s[i+k]==s[j+k])k++; height[rank[i]]=k; } } bool check(int k){ int cnt=1; for(int i=2;i<=n;i++){ if(height[i]>=k){ cnt++; }else cnt=1; if(cnt>=m)return 1; } return 0; } int erfen(int l,int r){ while(l<r){ int mid=(l+r+1)>>1; if(check(mid))l=mid; else r=mid-1; } return l; } int main(){ scanf("%d%d",&n,&m); for(int i=0;i<n;i++)scanf("%d",&s[i]); s[n++]=0; SA_init(); n--; height_init(); printf("%d\n",erfen(0,n)); return 0; }

+++++++++++++++++++++++++++++++++++++++++++

+本文作者:luyouqi233。               +

+歡迎訪問我的博客:http://www.cnblogs.com/luyouqi233/+

+++++++++++++++++++++++++++++++++++++++++++

POJ3261:Milk Patterns——題解