hdu 1403 后缀数组的应用

来源:互联网 发布:传智java视频教程 编辑:程序博客网 时间:2024/05/16 19:57

最近开始学后缀数组,不得不说这种结构真的很强大,但也很难学,看了好久都稀里糊涂的,现在又认真的看了一遍,总算有点想法了,然后从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 0x7f7f7fint 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;}


1 0
原创粉丝点击