ural 1297(后缀数组+RMQ)
题意:就是让你求一个字符串中的最长回文,如果有多个长度相等的最长回文,那就输出第一个最长回文。
思路:这是后缀数组的一种常见的应用,首先把原始字符串倒转过来,然后接在原始字符串的后面,中间用一个不可能出现的字符隔开。然后就用到后缀数组的性质了,
我们枚举每一个原始字符串中的字符以它为中心(分为奇数和偶数两种情况)进行查找,比如对于下标为i的字符,当回文串为奇数时,我们要求的就是i的后缀与2*n-i的
后缀的最长公共前缀了,然后根据height数组的性质就转化成求height[i+1]...height[2*n-i]中的最小值了,这时我们想到了用RMQ求区间最值了,对于不会RMQ的童
鞋可以先去学习下,再来做这道题,具体看代码实现:
代码实现:
#include<cstring> #include<cstdlib> #include<iostream> #include<cmath> using namespace std; #define N 2010 int ws1[N],wv[N],wa[N],wb[N]; int rank1[N],height[N],sa[N]; char str[N]; int a[N],n; int dp[N][25]; int min(int a,int b) { return a>b?b:a; } 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++) ws1[i]=0; for(i=0;i<n;i++) ws1[x[i]=r[i]]++; for(i=1;i<m;i++) ws1[i]+=ws1[i-1]; for(i=n-1;i>=0;i--) sa[--ws1[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++) ws1[i]=0; for(i=0;i<n;i++) ws1[wv[i]]++; for(i=1;i<m;i++) ws1[i]+=ws1[i-1]; for(i=n-1;i>=0;i--) sa[--ws1[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++; } } void calheight(int *r,int *sa,int n) { int i,j,k=0; for(i=1;i<=n;i++) rank1[sa[i]]=i; for(i=0;i<n;height[rank1[i++]]=k) for(k?k--:0,j=sa[rank1[i]-1];r[i+k]==r[j+k];k++) ; } void RMQ()//RMQ预处理 { int i,j; memset(dp,127,sizeof(dp)); for(i=1;i<=n*2+1;i++) dp[i][0]=height[i]; for(j=1;(1<<j)<=2*n+1;j++) for(i=1;i+(1<<j)-1<=2*n+1;i++) dp[i][j]=min(dp[i][j-1],dp[i+(1<<(j-1))][j-1]); } int lcp(int l,int r)//求最长公共前缀 { int a=rank1[l],b=rank1[r]; if(a>b) swap(a,b); a++; int t=(int)(log(double(b-a+1))/log(2.00)); return min(dp[a][t],dp[b-(1<<t)+1][t]); } int main() { int i,res,flag,max; while(scanf("%s",str)!=EOF) { max=0; n=strlen(str); for(i=0;i<n;i++) a[i]=(int)str[i]; a[n]=1; for(i=0;i<n;i++) a[i+n+1]=int(str[n-i-1]); a[2*n+1]=0; da(a,sa,n*2+2,123); calheight(a,sa,2*n+1); RMQ(); for(i=0;i<n;i++) { res=lcp(i,2*n-i)*2-1; if(max<res)//奇数时 { max=res; flag=i; } if(i>0)//偶数时 { res=lcp(i,2*n-i+1)*2; if(max<res) { max=res; flag=i; } } } if(max%2==1) for(i=flag-max/2;i<=flag+max/2;i++) printf("%c",str[i]); else for(i=flag-max/2;i<=flag+max/2-1;i++) printf("%c",str[i]); printf("\n"); } return 0; }
posted on 2013-08-10 10:01 后端bug开发工程师 阅读(1163) 评论(0) 编辑 收藏 举报