传送门:http://acm.hdu.edu.cn/showproblem.php?pid=1403
拖了这么久的后缀数组,终于准备学一学了。
做的第一题(套的第一个板子),虽然还是没看不懂是怎么实现的,但是还是可以套一套的。
suffix(i)表示[0,len)的字符串的一个后缀,下标为[i,len)。
后缀数组里,要求的是三个数组,sa[],rank[],height[]。
sa[i]表示的排名为i的后缀的位置。范围[1, len]。
rank[i]表示的是suffix(i)是排名第几。范围[0, len)。
sa[i]和rank[i]是一个相对关系。
height[i]表示suffix(sa[i-1])和suffix(sa[i])的最长公共前缀。范围[2, len]。
这题是求两个字符串的最长公共子串。而每个字符串的子串都是这个字符串的某个后缀的某个前缀。所以我们求出height数组以后,可以知道,公共前缀比较长的后缀的位置,肯定是排名相邻的,那么,我们只要把两个字符串拼接起来,中间拿一个没出现过的字符隔开。然后遍历height数组,如果两个后缀不属于同一个字符串,就更新答案,找出最长的。
后缀数组有两种实现,一个是倍增算法,复杂度O(nlogn)。一个是DC3算法,复杂度O(n)。
我参考了09年国家集训队论文,用的是倍增算法。
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <cstdlib>
#include <cctype>
#include <string>
#include <iostream>
#include <vector>
#include <map>
#include <set>
#include <stack>
#include <queue>
#include <ctime>
using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
#define pb push_back
#define mp make_pair
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define calm (l+r)>>1
const int INF = 2139062143;
const int maxn=200010;
struct SA{
int wa[maxn],wb[maxn],wv[maxn],ws[maxn];
int rank[maxn],height[maxn];
int cmp(int *r,int a,int b,int l){
return r[a]==r[b]&&r[a+l]==r[b+l];
}
void da(int *r,int *sa,int n,int m){
int i,j,p,*x=wa,*y=wb,*t;
for(i=0;i<m;i++) ws[i]=0;
for(i=0;i<n;i++) ws[x[i]=r[i]]++;
for(i=1;i<m;i++) ws[i]+=ws[i-1];
for(i=n-1;i>=0;i--) sa[--ws[x[i]]]=i;
for(j=1,p=1;p<n;j*=2,m=p){
for(p=0,i=n-j;i<n;i++) y[p++]=i;
for(i=0;i<n;i++) if(sa[i]>=j) y[p++]=sa[i]-j;
for(i=0;i<n;i++) wv[i]=x[y[i]];
for(i=0;i<m;i++) ws[i]=0;
for(i=0;i<n;i++) ws[wv[i]]++;
for(i=1;i<m;i++) ws[i]+=ws[i-1];
for(i=n-1;i>=0;i--) sa[--ws[wv[i]]]=y[i];
for(t=x,x=y,y=t,p=1,x[sa[0]]=0,i=1;i<n;i++)
x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;
}
return;
}
void calheight(int *r,int *sa,int n){
int i,j,k=0;
for(i=1;i<=n;i++) rank[sa[i]]=i;
for(i=0;i<n;height[rank[i++]]=k)
for(k?k--:0,j=sa[rank[i]-1];r[i+k]==r[j+k];k++);
return;
}
}rst;
char str[maxn];
int r[maxn],sa[maxn];
int main(){
//freopen("D://input.txt","r",stdin);
while(scanf("%s",str)!=EOF){
int len=strlen(str);
int mid=len;
str[len]='#';
scanf("%s",str+len+1);
len=strlen(str);
for(int i=0;i<=len;i++){
r[i]=str[i];
}
rst.da(r,sa,len+1,300);
rst.calheight(r,sa,len);
int ans=0;
for(int i=2;i<=len;i++){
if(ans<rst.height[i]){
if((sa[i]>mid&&sa[i-1]<mid)||(sa[i]<mid&&sa[i-1]>mid)){
ans=rst.height[i];
}
}
}
printf("%d\n",ans);
}
return 0;
}