UVA 12206 Stammering Aliens 后缀数组+二分

UVA 12206

题意：输入一个m，接下来输入一个仅包含小写字母的字符串，求字符串中至少出现m次的最长子串，输出两个数据，一个是最长子串长度，一个是这些子串中起始位置的最大值。

思路：比较水，很明显建立后缀数组，用二分找最长的子串并且用height数组去判断。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=4e4+100;
struct S_array
{
	int sa[maxn],t[maxn],t2[maxn],c[maxn],n;
	char s[maxn];
	void build_sa(int m)
	{
		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;
		}
	}
	int rank[maxn],height[maxn];
	void getHeight()
	{
		int i,j,k=0;
		for(i=0;i<n;i++)rank[sa[i]]=i;
		for(i=0;i<n;i++){
			if(k)k--;
			//if(rank[i]-1<0)
			//continue;
			int j=sa[rank[i]-1];
			while(s[i+k]==s[j+k])k++;
			height[rank[i]]=k;
		}
	}
}sa;
int check(int len,int minn)
{
	int ans=-1,flag=sa.sa[0],cnt=1;
	for(int i=1;i<sa.n;i++)
	{		
		if(sa.height[i]<len)
		{
			cnt=1;
			flag=sa.sa[i];
		}
		else
		{
			cnt++;	
			flag=max(flag,sa.sa[i]);
		}
		if(cnt>=minn)
		ans=max(ans,flag);
	}
	return ans;
}
void work(int minn)
{
	int L=0,R=sa.n+1;
	if(check(1,minn)==-1)
	{
		printf("none\n");
		return;
	}
	int m,ans;
	while(L<R)
	{
		m=(L+R)/2;
		ans=check(m,minn);
		if(ans!=-1)
		{
			if(L==m)
			break;
			L=m;
		}
		else
		R=m;
	}
	printf("%d %d\n",m,ans);
}
int main()
{
	int m;
	while(~scanf("%d",&m)&&m)
	{
		scanf("%s",sa.s);
		if(m==1)
		{
			printf("%d 0\n",strlen(sa.s));
			continue;
		}
		int t=strlen(sa.s);
		sa.s[t]='A';//字符串结尾为一个特殊的字符 
		sa.s[t+1]='\0';
		sa.n=t+1;
		sa.build_sa(180);
		sa.getHeight();
		work(m);
	}
}