POJ3261 求指定出现次数的可重叠的最长重复子串（后缀数组）

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cstring>
#include <string.h>
#include <queue>                //POJ3261 求指定出现次数的可重叠的最长重复子串（后缀数组）
#define maxn 20005
#define N 1000005
using namespace std;
int wa[maxn],wb[maxn],wv[maxn],Ws[N], Rank[maxn], height[maxn], sa[maxn];   //这里Ws的最大范围是这些数的最大值+1
int n, k, a[maxn];         
//以下为倍增算法，套模板
/*
height数组是从1到n的 
height数组是排名相邻的两个后缀的最长公共前缀（公共长度）
字符串中的两个子串的公共前缀想要最长，一定是在sa数组相邻的
*/
int cmp(int *r,int a,int b,int l)
{return r[a]==r[b]&&r[a+l]==r[b+l];}
void da(const int *r,int *sa,int n,int m)       //r为字符串(将其都转化为int）,n为字符串长度，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; 
}
//求height数组
void calheight(const 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 x)
{
	int t, j;
	for(j=2, t=0; j<=n; ++j)
	{
		if(height[j]>=x)         //height数组分组.<span style="font-family:Arial,Helvetica,sans-serif">相邻的两个,height值就表示最长公共前缀的长度，如果出现的次数达到k-1次，则return true</span><span style="font-family:Arial,Helvetica,sans-serif">
</span>		{
			t++;
			if(t==k-1)      
				return true;
		}
		else t=0;     //如果小于的话则需要将t重置为0
	}
	return false;
}
	
int main()
{
	int t, g, ans;
	while(scanf("%d%d", &n, &k)!=EOF)
	{
		for(t=g=0; t<n; ++t)
		{
			scanf("%d", a+t);
			a[t]++;       //每一项都+1，使得这个数列最小项为1
			if(a[t]>g)
				g=a[t];
		}
		a[t]=ans=0;   //末尾加入整个数列的最小值0
		da(a, sa, n+1, g+1);   //n+1包括自己加的最小值0
		calheight(a, sa, n);
		int l=0, r=n, mid;
		while(l<=r)       //二分法判定情况
		{
			mid=(l+r)>>1;
			if(check(mid))
			{
				ans=mid;
				l=mid+1;
			}
			else r=mid-1;
		}
		printf("%d\n", ans);
	}
	return 0;
}