poj 3261 Milk Patterns

3261 Milk Patterns

题意：这个题和1743很类似，只是这个题可以求的子串重复，并且次数要大于等于k

思路：还是利用height数组的性质就是两个排名相邻的两个后缀的最长瓜瓜前缀，如果要他们连续出现的只需要连续的height[i]的值都大于枚举的长度就可以了，因为这个题是可以重复的，所以没必要去判断这个组里的max(sa)-min(sa)的值是不是>l，在枚举长度的时候还是用二分。

/* * File: main.cpp * Author: Mi * * Created on 2011年4月27日, 下午4:16 */ #include <stdio.h> #include <string.h> #include <algorithm> #include <math.h> #include <map> #include <vector> #include <stdlib.h> #define MAX 1000005 using namespace std; /* * */ int wa[MAX],wb[MAX],ws[MAX],wv[MAX]; int r[MAX],sa[MAX]; int cmp(int *r,int a,int b,int l) { return r[a]==r[b]&&r[a+l]==r[b+l]; } void Da(int *r,int *sa,int n,int m) { int i,j,p,*x=wa,*y=wb,*t; for(i=0;i<m;i++) ws[i]=0; for(i=0;i<n;i++) ws[x[i]=r[i]]++; for(i=1;i<m;i++) ws[i]+=ws[i-1]; for(i=n-1;i>=0;i--) sa[--ws[x[i]]]=i; for(j=1,p=1;p<n;j*=2,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<n;i++) wv[i]=x[y[i]]; for(i=0;i<m;i++) ws[i]=0; for(i=0;i<n;i++) ws[wv[i]]++; for(i=1;i<m;i++) ws[i]+=ws[i-1]; for(i=n-1;i>=0;i--) sa[--ws[wv[i]]]=y[i]; for(t=x,x=y,y=t,p=1,x[sa[0]]=0,i=1;i<n;i++) x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++; } return ; } int rank[MAX],height[MAX]; void Callheight(int *r,int *sa,int n) { int i,j,k=0; for(i=1;i<=n;i++) rank[sa[i]]=i; for(i=0;i<n;height[rank[i++]]=k) for(k?k--:0,j=sa[rank[i]-1];r[i+k]==r[j+k];k++); return ; } int check(int n,int k,int mid) { int sum=1; for(int i=1;i<=n;i++) { if(height[i]>=mid)//看两个子串的最长公共子串是不是大于枚举的长度 { sum++;//统计连续的最长重复子串的个数 if(sum>=k) return 1; } else sum=1; } return 0; } int main(int argc, char** argv) { int n,k; scanf("%d%d",&n,&k); for(int i=0;i<n;i++) { scanf("%d",&r[i]); r[i]++; } r[n]=0; Da(r,sa,n+1,1000005); Callheight(r,sa,n); int l=1,r=n,mid; while(l<=r)//二分答案 { mid=(l+r)>>1; if(check(n,k,mid)) l=mid+1; else r=mid-1; } printf("%d/n",r); return 0; }