poj3261Milk Patterns（后缀数组）

题目链接：

http://poj.org/problem?id=3261

题目大意：

给一个字符串。问这个字符串中可重叠子串出现的次数超过K次的最大子串长度。

思路：

利用后缀数组height数组性质，我们可以知道相邻子串的公共前缀x，只要x的出现次数大于等于K就可以了。那么我们可以二分长度然后判断是否存在这种情况即可。

代码：

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
const int MAXN = 20005;
int rank[MAXN],height[MAXN];
int str[MAXN],s[MAXN];
int sa[MAXN],t[MAXN],t2[MAXN],c[MAXN];
void build_sa(int m,int n){
	int i, *x=t, *y=t2;
	for(i=0;i<m;i++)c[i]=0;
	for(i=0;i<n;i++)c[x[i]=s[i]]++;
	for(i=1;i<m;i++)c[i]+=c[i-1];
	for(i=n-1;i>=0;i--)sa[--c[x[i]]]=i;
	for(int k=1;k<=n;k<<=1){
		int p=0;
		for(i=n-k;i<n;i++)y[p++]=i;
		for(i=0;i<n;i++)if(sa[i]>=k)y[p++]=sa[i]-k;
		
		for(i=0;i<m;i++)c[i]=0;
		for(i=0;i<n;i++)c[x[y[i]]]++;
		for(i=0;i<m;i++)c[i]+=c[i-1];
		for(i=n-1;i>=0;i--)sa[--c[x[y[i]]]]=y[i];
		
		swap(x,y);
		p=1;x[sa[0]]=0;
		for(i=1;i<n;i++)
			x[sa[i]]=y[sa[i-1]]==y[sa[i]]&&y[sa[i-1]+k]==y[sa[i]+k]?p-1:p++;
		if(p>=n)break;
		m=p;
	}
}
void getHeight(int n){
	int i,j,k=0;
	for(i=0;i<=n;i++){rank[sa[i]]=i;}
	for(i=0;i<n;i++){
		if(k)k--;
		int j=sa[rank[i]-1];
		while(s[i+k]==s[j+k])k++;
		height[rank[i]]=k;
	}
}

int main()
{
	int vis[20005];
	int T,i,j;
	int n,K;
	scanf("%d%d",&n,&K);
		memset(vis,0,sizeof(vis));
		int sum=0;
		int ans=0;
		memset(str,0,sizeof(str));
		for(i=0;i<n;i++)
		scanf("%d",&str[i]);
		for(i=0;i<=n;i++)s[i]=str[i];
		sort(s,s+n);
		int siz=unique(s,s+n)-s;
		for(i=0;i<n;i++){
			str[i]=lower_bound(s,s+siz,str[i])-s+1;
		}
		for(i=0;i<=n;i++){
			s[i]=str[i];
		//	printf("%d* ",s[i]);
		}
		build_sa(MAXN,n+1);	
		getHeight(n);
		int l=1,r=n;
		while(l<=r){
			int cnt=1;
			int mid=l+r>>1;
			for(i=1;i<=n;i++){
				if(height[i]>=mid)cnt++;
				else cnt=1;
				if(cnt>=K)break;
			}
			if(cnt>=K)l=mid+1;
			else r=mid-1;	
			
		}
		printf("%d\n",r);
	
}