首页 > 代码库 > POJ 3261 Milk Patterns(后缀数组+二分答案)
POJ 3261 Milk Patterns(后缀数组+二分答案)
【题目链接】 http://poj.org/problem?id=3261
【题目大意】
求最长可允许重叠的出现次数不小于k的子串。
【题解】
对原串做一遍后缀数组,二分子串长度x,将前缀相同长度超过x的后缀分组,
如果存在一个大小不小于k的分组,则说明答案可行,分治得到最大可行解就是答案。
【代码】
#include <cstdio>#include <cstring>#include <vector>using namespace std;const int N=1000010,M=1000010;int n,rank[N],sa[N],h[N],tmp[N],cnt[N],ans;int s[N];void suffixarray(int n,int m){ int i,j,k;n++; for(i=0;i<2*n+5;i++)rank[i]=sa[i]=h[i]=tmp[i]=0; for(i=0;i<m;i++)cnt[i]=0; for(i=0;i<n;i++)cnt[rank[i]=s[i]]++; for(i=1;i<m;i++)cnt[i]+=cnt[i-1]; for(i=0;i<n;i++)sa[--cnt[rank[i]]]=i; for(k=1;k<=n;k<<=1){ for(i=0;i<n;i++){ j=sa[i]-k; if(j<0)j+=n; tmp[cnt[rank[j]]++]=j; }sa[tmp[cnt[0]=0]]=j=0; for(i=1;i<n;i++){ if(rank[tmp[i]]!=rank[tmp[i-1]]||rank[tmp[i]+k]!=rank[tmp[i-1]+k])cnt[++j]=i; sa[tmp[i]]=j; }memcpy(rank,sa,n*sizeof(int)); memcpy(sa,tmp,n*sizeof(int)); if(j>=n-1)break; }for(j=rank[h[i=k=0]=0];i<n-1;i++,k++) while(~k&&s[i]!=s[sa[j-1]+k])h[j]=k--,j=rank[sa[j]+1];}vector<int> v[N]; int k; bool check(int x){ int cnt=-1; for(int i=1;i<=n;i++){ if(h[i]<x)v[++cnt].clear(); v[cnt].push_back(i); }for(int i=0;i<=cnt;i++)if(v[i].size()>=k)return 1; return 0;}int main(){ while(~scanf("%d%d",&n,&k)){ for(int i=0;i<n;i++)scanf("%d",&s[i]); suffixarray(n,M); int l=0,r=n,ans=0; while(l<=r){ int mid=(l+r)>>1; if(check(mid))ans=mid,l=mid+1; else r=mid-1; }printf("%d\n",ans); }return 0;}
POJ 3261 Milk Patterns(后缀数组+二分答案)
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。