bzoj2251: [2010Beijing Wc]外星联络 后缀数组

后缀数组找出排名，然后强上O(n^2)枚举即可。

#include<bits/stdc++.h>
using namespace std;
int str[3010],n,p[3010],cnt[3010],temp[3010],sa[3010],rank[3010],f[3010][3010],w[9100000],height[3010],tot=0;
char s[3010];
void init()
{
	scanf("%d",&n);scanf("%s",s);
	for(int i=0;i<n;i++)str[i]=s[i]-'0';
}
bool epu(int x,int y,int l){return rank[x]==rank[y]&&rank[x+l]==rank[y+l];}
void doubling()
{
	rank[n]=str[n]=-1;
	for(int i=0;i<n;i++)sa[i]=i,rank[i]=str[i];
	for(int i,l=0,pos=0,sig=1;pos<n-1;sig=pos)
	{
		for(i=n-l,pos=0;i<n;i++)p[pos++]=i;
		for(i=0;i<n;i++)if(sa[i]>=l)p[pos++]=sa[i]-l;
		memset(cnt,0,sizeof(cnt));
		for(i=0;i<n;i++)cnt[rank[p[i]]]++;
		for(i=1;i<=sig;i++)cnt[i]+=cnt[i-1];
		for(i=n-1;i>=0;i--)sa[--cnt[rank[p[i]]]]=p[i];
		for(temp[sa[0]]=pos=0,i=1;i<n;i++)
		{
			if(!epu(sa[i],sa[i-1],l))pos++;
			temp[sa[i]]=pos;
		}
		for(i=0;i<n;i++)rank[i]=temp[i];
		if(!l)l=1;else l<<=1;
	}
}
void pre()
{
	int i=0,k=0,j;
	for(i=0;i<n;i++)
	{
		if(k)k--;
		if(rank[i]==0)continue;
		for(j=sa[rank[i]-1];str[j+k]==str[i+k];)
			k++;
		height[rank[i]]=k;
	}
}
void work()
{
	int i,j;
	for(i=0;i<n;i++)
	{
		for(j=0;j<height[i];j++){f[i][j]=f[i-1][j];w[f[i][j]]++;}
		for(j=height[i];j<n;j++){f[i][j]=++tot;w[f[i][j]]++;}
	}
	for(int i=1;i<=tot;i++)if(w[i]>1)printf("%d\n",w[i]);
}
int main()
{
	init();
	doubling();
	pre();
	work();
	return 0;
}

　　

转载于:https://www.cnblogs.com/mybing/p/8530860.html