poj3261Milk Patterns

最新推荐文章于 2019-06-02 00:59:00 发布

原创最新推荐文章于 2019-06-02 00:59:00 发布 · 216 阅读

0 ·

CC 4.0 BY-SA版权

后缀数组专栏收录该内容

7 篇文章

订阅专栏

本文介绍了一种求解最长重复子串的算法实现，通过后缀数组和高度数组结合二分查找技巧，高效找到字符串中重复k次的最长子串。文章提供了完整的C++代码示例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目大意：求重复k次的最长重复子串。

其实就是拿来练练手。毕竟学了后缀有一年了。

求出height，然后二分答案。

#include<cstdio>
#include<cstdlib>
#include<cstdlib>
const int N=20010;
const int M=1000002;
int ws[M],wy[N],wx[N],wv[N];
int cmp(int *r,int x,int x1,int ln)
{
	return r[x]==r[x1]&&r[x+ln]==r[x1+ln];
}
void da(int *r,int *sa,int n,int m)
{
	int *x=wx,*y=wy,*t,i,j,p;
	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,i=1,x[sa[0]]=0,p=1;i<n;i++)
		x[sa[i]]=cmp(y,sa[i],sa[i-1],j)?p-1:p++;
	}
}
int rank[N],height[N];
void calheight(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;
}
bool check(int mid,int n,int k)
{
	int s=1;
	for(int i=1;i<=n;i++)
	{
		if(height[i]>=mid)
		{
			s++;
			if(s>=k) return 1;
		}
		else s=1;
	}
	return 0;
}
int n,k;
int r[N],sa[N];
int main()
{
	scanf("%d%d",&n,&k);
	for(int i=0;i<n;i++)
	{
		scanf("%d",&r[i]);
		r[i]++;
	}
	da(r,sa,n+1,1000002);
	calheight(r,sa,n);
	int l=1,r=n;
	while(l<=r)
	{
		int mid=(l+r)>>1;
		if(check(mid,n,k))
		{
			l=mid+1;
		}
		else r=mid-1;
	}
	printf("%d\n",r);
}