【POJ2774】后缀数组

【POJ2774】

题目大意

---

• 给定两个字符串a，b求它们的最长公共字串
• len(a),len(b) <= 1e5

 

分析

---

• 构造一个串s，s=a+'$'+b，这样可以保证s的后缀的LCP不会跨过a，b
• 那么公共字串就转化成了s的后缀的LCP
• 可以发现后缀排名离得越近的LCP就越长，那么答案就产生在height数组中
• 求出s的height[]和sa[]
• 答案就是MAX{ height[i] }  排名为i的后缀和排名为i-1的后缀要分别属于a,b或b,a最开始忘记判断了

Code

---

#include<bits/stdc++.h>
#define rep(i,a,b) for(int i=(a);i<=(b);++i)
#define per(i,a,b) for(int i=(a);i>=(b);--i)
using namespace std;
const int N=1e6+100;
int rk[N],sa[N],tmp[N],bin[N],h[N],height[N],n,la,lb,ans;
string s,a,b;
void SA(){
	int sz=max(n,130);
	rep(i, 0,   n-1 )	rk[i]=s[i];
	rep(i, 0,   n-1 ) 	++bin[rk[i]];
	rep(i, 1,   sz-1)	bin[i]+=bin[i-1];
	per(i, n-1, 0   ) 	sa[--bin[rk[i]]]=i;
	for(int j=1;j<=n;j*=2)
	{
		int p=0;
		per(i,	n-1, n-j )   tmp[p++]=i;
		rep(i,	0,	 n-1 )	 if(sa[i]-j>=0)tmp[p++]=sa[i]-j;
		rep(i,	0,	 sz-1) 	 bin[i]=0;
		rep(i,	0,	 n-1 )   ++bin[rk[i]];
		rep(i,	1,	 sz-1) 	 bin[i]+=bin[i-1];
		per(i,	n-1, 0   ) 	 sa[--bin[rk[tmp[i]]]]=tmp[i];
		p=tmp[sa[0]]=0;
		rep(i,1,n-1){
			int v0=sa[i-1],v1=sa[i],v00,v01;
			if (v0 + j < n) v00 = rk[v0 + j]; else v00 = -1 ;
        	        if (v1 + j < n) v01 = rk[v1 + j]; else v01 = -1 ;
			if(rk[v0]==rk[v1] && v00==v01) tmp[v1]=p;
			else tmp[v1]=++p;
		}
		rep( i, 0, n-1  )    rk[i]=tmp[i];
	}
}
int getheight() {
    int k=0;
    rep(i,0,n-1) rk[sa[i]]=i;
    rep(i,0,n-1) {
        if (rk[i]==0) continue;
        if (k) --k;
        int j=sa[rk[i]-1];
        while (j+k<n && i+k<n && s[i+k]==s[j+k]) ++k;
        h[i]=height[rk[i]]=k;
    }
}
int main()
{
	cin>>a>>b;la=a.size();lb=b.size();
	s=a+'$'+b; n=s.size();
	SA(); getheight();
	rep(i,1,n-1)if(sa[i]<la&&sa[i-1]>la||sa[i]>la&&sa[i-1]<la)ans=max(ans,height[i]);
	cout<<ans;
	return 0; 
}