Poj 3693 Maximum repetition substring|后缀数组|st表

这一定是我做的最恶心的一道后缀数组的题

简直恶心哭了

具体做法可以参见罗穗骞的论文

说一下字典序最小怎么找:

找出所有重复最多的子串长度

枚举sa数组最先找到的那个满足条件的 就是字典序最小的

ps:

1,st表查询的时候 左边的rank要加1!

#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<queue>
#include<vector>
#include<set>
#include<map>
#include<iostream>
#include<algorithm>
#define T 100010
using namespace std;
char s[T];
int t1[T],t2[T],sa[T],cc[T],rank[T],height[T];
int st[T][17],z[T];
int len,ans,cnt,flag,num=0;
bool cmp(int *y,int a,int b,int k)
{
	int a1=y[a],b1=y[b];
	int a2=a+k>=len?-1:y[a+k];
	int b2=b+k>=len?-1:y[b+k];
	return a1==b1&&a2==b2;
}
void make_sa()
{
	int *x=t1,*y=t2,m=333;
	for(int i=0;i<m;i++)cc[i]=0;
	for(int i=0;i<len;i++)++cc[x[i]=s[i]];
	for(int i=1;i<m;i++)cc[i]+=cc[i-1];
	for(int i=len-1;~i;i--)sa[--cc[x[i]]]=i;
	for(int k=1;k<=len;k<<=1)
	{
		int p=0;
		for(int i=len-k;i<len;i++)y[p++]=i;
		for(int i=0;i<len;i++)if(sa[i]>=k)y[p++]=sa[i]-k;
		for(int i=0;i<m;i++)cc[i]=0;
		for(int i=0;i<len;i++)++cc[x[y[i]]];
		for(int i=1;i<m;i++)cc[i]+=cc[i-1];
		for(int i=len-1;~i;i--)sa[--cc[x[y[i]]]]=y[i];
		swap(x,y);m=1;x[sa[0]]=0;
		for(int i=1;i<len;i++)x[sa[i]]=cmp(y,sa[i],sa[i-1],k)?m-1:m++;
		if(m>=len)return;
	}
}
void make_height()
{
	int k=0;
	for(int i=0;i<len;i++)rank[sa[i]]=i;
	for(int i=0;i<len;i++)
	{
		if(!rank[i])continue;
		int j=sa[rank[i]-1];
		if(k)k--;
		while(s[i+k]==s[j+k])k++;
		height[rank[i]]=k;
	}
}
void make_st()
{
	int w=log2(len);
	st[0][0]=T;
	for(int i=1;i<len;i++)st[i][0]=height[i];//cout << i <<" "<<height[i]<<endl;
	for(int k=1;k<=w;k++)
	    for(int i=0;i<len;i++)
	    {
	    	if(i+(1<<k)>=len)break;
	    	st[i][k]=min(st[i][k-1],st[i+(1<<k-1)][k-1]);
	    }
}
int ask(int l,int r)
{
	if(l>r)swap(l,r);l++;
	int k=log2(r-l+1);
	return min(st[l][k],st[r-(1<<k)+1][k]);
}
int main()
{
	while(scanf("%s",s)&&s[0]!='#')
	{
		len=strlen(s);
		make_sa();
		make_height();
		make_st();
		ans=cnt=0;
		for(int l=1;l<=len;l++)
		    for(int i=l;i<len;i+=l)
		    {
		    	int k=ask(rank[i],rank[i-l]); 
		    	int m=l-k%l;
		    	if(i>l&&k%l)k=max(k,ask(rank[i-l-m],rank[i-m]));
		    	k=k/l+1;
		    	if(k>ans)ans=k,cnt=1,z[1]=l;
		    	else if(k==ans&&z[cnt]!=l)z[++cnt]=l;
		    }
		flag=1;
		for(int i=0;i<len&&flag;i++)
		    for(int j=1;j<=cnt&&flag;j++)
		        if(ask(rank[sa[i]],rank[sa[i]+z[j]]) >= (ans-1)*z[j])
		        {
		        	//if(sa[i]+z[j]*ans>=len)continue;
		        	flag=0;
		        	printf("Case %d: ",++num);
		        	s[sa[i]+z[j]*ans]='\0';
		        	printf("%s\n",s+sa[i]);
		        }
	}
}



你可能感兴趣的:(后缀数组,st表)