bzoj1717: [Usaco2006 Dec]Milk Patterns 产奶的模式[后缀数组]题解②

1717: [Usaco2006 Dec]Milk Patterns 产奶的模式

Time Limit: 5 Sec   Memory Limit: 64 MB
Submit: 1011   Solved: 558
[ Submit][ Status][ Discuss]

Description

农夫John发现他的奶牛产奶的质量一直在变动。经过细致的调查，他发现：虽然他不能预见明天产奶的质量，但连续的若干天的质量有很多重叠。我们称之为一个“模式”。 John的牛奶按质量可以被赋予一个0到1000000之间的数。并且John记录了N(1<=N<=20000)天的牛奶质量值。他想知道最长的出现了至少K(2<=K<=N)次的模式的长度。比如1 2 3 2 3 2 3 1 中 2 3 2 3出现了两次。当K=2时，这个长度为4。

Input

* Line 1: 两个整数 N,K。

* Lines 2..N+1: 每行一个整数表示当天的质量值。

Output

* Line 1: 一个整数：N天中最长的出现了至少K次的模式的长度

Sample Input

8 2
1
2
3
2
3
2
3
1

Sample Output

4

HINT

Source

Gold

hash解法链接： 【bzoj1717】[Usaco2006 Dec]Milk Patterns 产奶的模式 （二分+hash）.

终于知道后缀数组怎么做这题了

二分长度，将height>=mid的i分为一组，判断是否有一组的元素个数>=k即可

代码：

#include<iostream>
#include<algorithm>
#include<cstring>
#include<string>
#include<cstdio>
#include<climits>
#include<queue>
#include<stack>
#include<map>
#include<set>
#define N 800010
using namespace std;
char ch[N];int k,ans;
int n,a[N],hash[N],num[N],tot;
int wa[N],wb[N],wv[N],cnt[N];
int sa[N],rank[N],height[N];
void cal_sa(int *r,int n,int m)
{
	int *x=wa,*y=wb;
	for(int i=0;i<m;i++)cnt[i]=0;
	for(int i=0;i<n;i++)cnt[x[i]=r[i]]++;
	for(int i=1;i<m;i++)cnt[i]+=cnt[i-1];
	for(int i=n-1;i>=0;i--)sa[--cnt[r[i]]]=i;
	
	for(int j=1;j<n;j<<=1)
	{
		int p=0;
		for(int i=n-j;i<n;i++)y[p++]=i;
		for(int i=0;i<n;i++)if(sa[i]>=j)y[p++]=sa[i]-j;
		for(int i=0;i<n;i++)wv[i]=x[y[i]];
		
		for(int i=0;i<m;i++)cnt[i]=0;
		for(int i=0;i<n;i++)cnt[wv[i]]++;
		for(int i=1;i<m;i++)cnt[i]+=cnt[i-1];
		for(int i=n-1;i>=0;i--)sa[--cnt[wv[i]]]=y[i];
		p=1;swap(x,y);x[sa[0]]=0;
		for(int i=1;i<n;i++)
			x[sa[i]]=y[sa[i]]==y[sa[i-1]]&&y[sa[i]+j]==y[sa[i-1]+j]?p-1:p++;
		if(p>=n)break;m=p;
	}
}
void cal_h(int *r,int n)
{
	int i=0,j=0,k=0;
	for(int i=0;i<n;i++)
		rank[sa[i]]=i;
	for(i=0,j=0,k=0;i<n;i++)
	{
		if(!rank[i])continue;
		if(k)k--;j=sa[rank[i]-1];
		while(r[j+k]==r[i+k]&&i+k<n&&j+k<n)k++;
		height[rank[i]]=k;
	//	printf("! %d %d ! @%d %d@\n",rank[i],height[rank[i]],sa[rank[i]],sa[rank[i]-1]);
	}
}
int find(int x)
{
	int l=1,r=tot,ret;
	while(l<=r)
	{
		int mid=(l+r)>>1;
		if(hash[mid]<=x)
			l=mid+1,ret=mid;
		else r=mid-1;
	}return ret;
}
bool check(int len)
{
	int cnt=1;
	for(int i=1;i<=n;i++)
	{
		if(height[i]<len)
		{
			if(cnt>=k)return true;
			cnt=0;
		}
		cnt++;
	}if(cnt>=k)return true;
	return false;
}
void solve()
{
	int l=0,r=n,ret;
	while(l<=r)
	{
		int mid=(l+r)>>1;
		if(check(mid))
			l=mid+1,ret=mid;
		else r=mid-1;
	}ans=ret;
}
int main()
{
//	freopen("poj3261.in","r",stdin);
//	freopen("poj3261.out","w",stdout);
	scanf("%d%d",&n,&k);
	for(int i=0;i<n;i++)
	{
		scanf("%d",&a[i]);
		num[i]=a[i];
	}sort(num,num+n);
	for(int i=0;i<n;i++)
	{
		if(i&&num[i]==num[i-1])continue;
		hash[++tot]=num[i];
	}for(int i=0;i<n;i++)a[i]=find(a[i]);a[n]=0;
	cal_sa(a,n,tot+1);cal_h(a,n);solve();
	printf("%d\n",ans);
}