【模板】后缀数组

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
//Accepted???
using namespace std;
const int N=2e5+5;
const int Maxchar=26;
char s[N],a[N];
int n,m,nn,num,x[N],y[N],c[N],sa[N];
int h[N],height[N],rk[N];
int f[N][20],res;
void solve() {
	m=Maxchar;
	for(int i=1;i<=n;i++) c[x[i]=s[i]-'a'+1]++;
	for(int i=2;i<=m;i++) c[i]+=c[i-1];
	for(int i=n;i>=1;i--) sa[c[x[i]]--]=i; 
	for(int k=1;k<=n;k<<=1) {
		num=0;
		for(int i=n-k+1;i<=n;i++) y[++num]=i;
		for(int i=1;i<=n;i++) if(sa[i]>k) y[++num]=sa[i]-k;
		for(int i=1;i<=m;i++) c[i]=0;
		for(int i=1;i<=n;i++) c[x[i]]++;
		for(int i=2;i<=m;i++) c[i]+=c[i-1];
		for(int i=n;i>=1;i--) sa[c[x[y[i]]]--]=y[i];
		for(int i=1;i<=n;i++) y[i]=x[i],x[i]=0;
		x[sa[1]]=1;num=1;
		for(int i=2;i<=n;i++) x[sa[i]]=(y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k])?num:++num;
		if(m==n) break;
		m=num;
	}
} 
void solve2() {
	int k=0;
	for (int i=1; i<=n; ++i) rk[sa[i]]=i;
	for (int i=1; i<=n; ++i) {
		if (rk[i]==1) continue;//第一名height为0
		if (k) --k;//h[i]>=h[i-1]-1;
		int j=sa[rk[i]-1];
		while (j+k<=n && i+k<=n && s[i+k]==s[j+k]) ++k;
		height[rk[i]]=k;//h[i]=height[rk[i]];
	}
	for(int i=1;i<=n;i++) f[i][0]=height[i];
	for(int j=1;j<20;j++) {
		for(int i=1;i<=n-(1<<j)+1;i++) {
			f[i][j]=min(f[i][j-1],f[i+(1<<(j-1))][j-1]);
		} 
	}
}
int ask(int x,int y) {
	int k=int(log(1.0*(y-x+1))/log(2.0));
	return min(f[x][k],f[y-(1<<k)+1][k]); 
}
void solve3() {
	int k1=0,k2=0;
	for(int i=1;i<=n;i++) {
//		printf("%d ",sa[i]);
		if(sa[i]<=nn) {
			if(k2) res=max(res,ask(k2+1,i));
			k1=i;
		}
		else {
			if(k1) res=max(res,ask(k1+1,i));
			k2=i;
		}
	}
}
//思路:求出两个数的后缀数组,然后按字典序排序.
//根据LCP的性质,a,b的lcp为[a,b]中height的最小值,那么枚举a串,找到距离最近的b串求lcp的贡献即可.
int main() {
	memset(f,0x3f,sizeof(f));
	scanf("%s",s+1); n=strlen(s+1); nn=n;
	scanf("%s",a+1); int len=strlen(a+1);
	for(int i=1;i<=len;i++) s[n+i]=a[i];
	n+=len; 
	solve();
	solve2();
	solve3();
	printf("%d",res);
}
posted @ 2020-12-29 17:29  仰望星空的蚂蚁  阅读(13)  评论(0)    收藏  举报  来源