2018.11.24 poj3693Maximum repetition substring(后缀数组)

时间:2023-03-10 03:20:53
2018.11.24 poj3693Maximum repetition substring(后缀数组)

传送门

后缀数组好题。


考虑枚举循环节长度lenlenlen。

然后考虑枚举循环节的起点来更新答案。

但是直接枚举每次O(n)O(n)O(n)。

考虑枚举len∗k+1len*k+1len∗k+1作为起点。

然后用len∗(k−1)+1len*(k-1)+1len∗(k−1)+1和len∗k+1len*k+1len∗k+1之间的某一段有可能使循环节长度再加一。

代码:

#include<iostream>
#include<cstdio>
#include<cstring>
#define ri register int
using namespace std;
const int N=1e5+5;
char s[N];
int sa2[N],cnt[N],n,Log[N],m,sa[N],rk[N],ht[N],st[N][17],pos[N],top=0;
inline void Sort(){
	for(ri i=1;i<=m;++i)cnt[i]=0;
	for(ri i=1;i<=n;++i)++cnt[rk[i]];
	for(ri i=2;i<=m;++i)cnt[i]+=cnt[i-1];
	for(ri i=n;i;--i)sa[cnt[rk[sa2[i]]]--]=sa2[i];
}
inline void getsa(){
	for(ri i=1;i<=n;++i)rk[i]=s[i]-'a'+1,sa2[i]=i;
	m=130,Sort();
	for(ri w=1,p=0;m!=n;p=0,w<<=1){
		for(ri i=n-w+1;i<=n;++i)sa2[++p]=i;
		for(ri i=1;i<=n;++i)if(sa[i]>w)sa2[++p]=sa[i]-w;
		Sort(),swap(rk,sa2),rk[sa[1]]=p=1;
		for(ri i=2;i<=n;++i)rk[sa[i]]=(sa2[sa[i]]==sa2[sa[i-1]]&&sa2[sa[i]+w]==sa2[sa[i-1]+w])?p:++p;
		m=p;
	}
	for(ri i=1,j,k=0;i<=n;ht[rk[i++]]=k)for(k?--k:k,j=sa[rk[i]-1];s[i+k]==s[j+k];++k);
	for(ri i=1;i<=n;++i)st[i][0]=ht[i];
	for(ri j=1;j<=16;++j)for(ri i=1;i+(1<<j)<=n;++i)st[i][j]=min(st[i][j-1],st[i+(1<<(j-1))][j-1]);
}
inline int rmq(int l,int r){return min(st[l][Log[r-l+1]],st[r+1-(1<<Log[r-l+1])][Log[r-l+1]]);}
inline int Query(int l,int r){return rmq(min(rk[l],rk[r])+1,max(rk[l],rk[r]));}
inline void query(){
	int tim=1,ans=1,len=0;
	for(ri T=1;T<n;++T){
		for(ri l=1,r=T+1;r<=n;l+=T,r+=T){
			int L=Query(l,r),stp=L/T+1,pl=l-(T-L%T),pr=pl+T;
			if(pl>0&&L%T)if(Query(pl,pr)>=L)++stp;
			if(stp>tim)tim=stp,pos[top=1]=T;
			else if(stp==tim)pos[++top]=T;
		}
	}
	for(ri i=1;!len&&i<=n;++i)for(ri j=1,T=pos[j];j<=top;T=pos[++j])if(Query(sa[i],sa[i]+T)>=(tim-1)*T){ans=sa[i],len=T;break;}
	for(ri i=ans,j=1;j<=len*tim;++j,++i)printf("%c",s[i]);
}
int main(){
	for(ri i=2;i<=100000;++i)Log[i]=Log[i>>1]+1;
	for(ri tt=1;;++tt,puts(""),top=0){
		scanf("%s",s+1),n=strlen(s+1);
		if(s[1]=='#')break;
		printf("Case %d: ",tt),getsa(),query();
	}
	return 0;
}