poj3261（后缀数组）

题意：给出一串长度为n的字符，再给出一个k值，要你求重复次数大于等于k次的最长子串长度........

思路：其实也非常简单，直接求出height值，然后将它分组，二分答案......结果就出来了.......

#include<iostream>
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
#define maxx 110000
int wa[maxx],wb[maxx],wsf[maxx],wv[maxx];
int sa[maxx],rank[maxx],s[maxx],height[maxx];
struct node
{
	int num,x;
}str[maxx];
int cmp1(const node a,const node b)
{
	if(a.x<b.x)
	return 1;
	else
	return 0;
}
int cmp(int *r,int a,int b,int k)
{
	return r[a]==r[b]&&r[a+k]==r[b+k];
}
void getsa(int *r,int *sa,int n,int m)
{
	int i,j,p,*x=wa,*y=wb,*t;
	for(i=0;i<m;i++)  wsf[i]=0;
	for(i=0;i<n;i++)  wsf[x[i]=r[i]]++;
	for(i=1;i<m;i++)  wsf[i]+=wsf[i-1];
	for(i=n-1;i>=0;i--)  sa[--wsf[x[i]]]=i;
	j=1;
	p=1;
	for(;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++)  wsf[i]=0;
		for(i=0;i<n;i++)  wsf[wv[i]]++;
		for(i=1;i<m;i++)  wsf[i]+=wsf[i-1];
		for(i=n-1;i>=0;i--)  sa[--wsf[wv[i]]]=y[i];
		t=x;
		x=y;
		y=t;
		x[sa[0]]=0;
		for(p=1,i=1;i<n;i++)   x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;
	}
}
void getheight(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;i++)
	{
		if(k)
		k--;
		else
		k=0;
		j=sa[rank[i]-1];
		while(r[i+k]==r[j+k])
		k++;
		height[rank[i]]=k;
	}
}
int deal(int n,int mid,int k)
{
	int flag=1;                     //要注意，初始值是1.....应该一个height值，是两个字符的lcp，i与i-1的....所以初始值应该从1开始....
	for(int i=2;i<=n;i++)
	{
		if(height[i]>=mid)
		{
			flag++;
			if(flag>=k)
			return 1;
		}
		else  flag=1;
	}
	return 0;
}
int main()
{
	int n,k;
	while(scanf("%d%d",&n,&k)>0)
	{
		int zd=0;
		for(int i=0;i<n;i++)
		{
			scanf("%d",&str[i].x);
			str[i].num=i;
		}
		sort(str,str+n,cmp1);
		int j=1;
		s[str[0].num]=j;
		for(int i=1;i<n;i++)
		{
			if(str[i].x==str[i-1].x)
			s[str[i].num]=j;
			else
			{
				j++;
				s[str[i].num]=j;
			}
		}
		s[n]=0;
		//for(int i=0;i<n;i++)
		//printf("%d ",s[i]);
		getsa(s,sa,n+1,j+10);
		getheight(s,sa,n);
		int left=0,right=n,count=0,mid;
		while(left<=right)
		{
			mid=(left+right)/2;
			if(deal(n,mid,k))
			{
				if(count<mid)
				count=mid;
				//printf("%d\n",mid);
				left=mid+1;
			}
			else  right=mid-1;
		}
		printf("%d\n",count);
	}
	return 0;
}