最近开始学后缀数组,不得不说这种结构真的很强大,但也很难学,看了好久都稀里糊涂的,现在又认真的看了一遍,总算有点想法了,然后从HDU找了一道题来小试一下,没想到轻松过。
题目是求两个字符串的最长公共子串,思想是连接两个字符串,中间用一个特殊字符隔开,然后再遍历height数组,找最大值,且满足s[i]与s[i-1]属于不同的字符串
下面便是我的代码:
/*******************************************************************************/
/* OS : Linux fc20.x86_64 #1 SMP Tue Dec UTC 2013 x86_64 GNU/Linux
* Compiler : 4.8.2 20131212 (Red Hat 4.8.2-7) (GCC)
* Encoding : UTF8
* Date : 2014-04-02
* All Rights Reserved by alop.
*****************************************************************************/
/* Description: ***************************************************************
*****************************************************************************/
/* Analysis: ******************************************************************
*****************************************************************************/
/*****************************************************************************/
#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<string>
using namespace std;
#define N 200010
#define INF 0x7f7f7f
int WA[N],WB[N],WV[N],WS[N];
int Rank[N],height[N];
int cmp(int *r,int a,int b,int l)
{
return r[a]==r[b]&&r[a+l]==r[b+l];
}
void da(char *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(p=1,j=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(char *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;
}
char str[N];
int sa[N];
int main()
{
while(scanf("%s",str)!=EOF)
{
int len=strlen(str);
str[len]='0';
scanf("%s",str+len+1);
int siz=strlen(str);
da(str,sa,siz,128);
calheight(str,sa,siz);
int ans=0;
for(int i=1; i<siz; i++)
{
if(ans<height[i])
{
if((sa[i]>len&&sa[i-1]<len)||(sa[i]<len&&sa[i-1]>len))
ans=height[i];
}
}
cout<<ans<<endl;
}
return 0;
}