BZOJ1717 产奶的模式

题目大意：求至少出现k次的可重复子串的最大长度。

解析：后缀数组求height值，二分出长度判断：如果连续k-1个height值都大于等于mid，就满足，否则不满足。

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int MAX=1000010;
char s[MAX],t[MAX];
int x[MAX],y[MAX],c[MAX*10],sa[MAX];
int n,m=200,k,cnt=0,ans;
int cmp(int *r, int a, int b, int l) {return r[a]==r[b]&&r[a+l]==r[b+l];}
void build_sa()
{
	for (int i=0;i<m;i++) c[i]=0;
	for (int i=0;i<n;i++) c[x[i]=s[i]]++;
	for (int i=1;i<m;i++) c[i]+=c[i-1];
	for (int i=n-1;i>=0;i--) sa[--c[x[i]]]=i;
	for (int k=1;k<=n;k<<=1)
	{
		int p=0;
		for (int i=n-k;i<n;i++)	y[p++]=i;
		for (int i=0;i<n;i++)	
			if (sa[i]>=k) 
				y[p++]=sa[i]-k;
		for (int i=0;i<m;i++) c[i]=0;
		for (int i=0;i<n;i++) c[x[i]]++;
		for (int i=1;i<m;i++) c[i]+=c[i-1];
		for (int i=n-1;i>=0;i--) sa[--c[x[y[i]]]]=y[i];
		swap(x,y); p=1; x[sa[0]]=0;
		for (int i=1;i<n;i++)
			x[sa[i]]=cmp(y,sa[i-1],sa[i],k)?p-1:p++;
		if(p>n) break;
		m=p;	
	}
}
int rank[MAX],height[MAX];
void build_height()
{
    for (int i=0;i<n;++i) rank[sa[i]]=i;
    int k=0;height[0]=0;
    for (int i=0;i<n;++i)
    {
        if (!rank[i]) continue;
        if (k) --k;
        int j=sa[rank[i]-1];
        while (i+k<n&&j+k<n&&s[i+k]==s[j+k]) ++k;
        height[rank[i]]=k;
    }
}
bool check(int mid)
{
	int num=0,minn=0;
	for(int i=0;i<n;i++)
	{
		if(height[i]<mid)
		{
			num=0;
			continue;
		}
		else if(height[i]>=mid)
			num++;
		if(num==cnt-1)
			return true;
	}
	return false;
}
int main()
{
	scanf("%d%d",&n,&cnt);
	for (int i=1;i<=n;i++)
	{
		int x;
		scanf("%d",&x);
		s[i-1]=x+48;
	}
	build_sa();
	build_height();
	int l=1,r=n;
	while(l<=r)
	{
		int mid=(l+r)/2;
		if(check(mid))
		{
			l=mid+1;
			ans=mid;
		}	
		else
			r=mid-1;
	}
	printf("%d",ans);
}