ural 1297 Palindrome ( 后缀数组 最长回文子串 )

题意 : 计算一个字符串的最长回文子串

思路 : 首先将字符串反转之后加到原来字符串的后面，中间用一个没有出现过的字符隔开。求最长回文子串我们就可以枚举回文串的中心，那么以这个字符为中心的最长回文子串可以利用以这个字符为第一个字符的后缀和这个字符在反转的串中的对应位置的后缀的LCP。 注意求解的时候要分奇偶去考虑。

#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;

#define maxn 2005
#define maxm 2005

int wa[maxn],wb[maxn],wv[maxn],ws[maxn];
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(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;
}
int Rank[maxn],height[maxn];
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;
}
int RMQ[maxn];
int mm[maxn];
int best[20][maxn];
void initRMQ(int n)
{
     int i,j,a,b;
	 for(int i=1;i<=n;i++)
		RMQ[i] = height[i] ;
     for(mm[0]=-1,i=1;i<=n;i++)
     mm[i]=((i&(i-1))==0)?mm[i-1]+1:mm[i-1];
     for(i=1;i<=n;i++) best[0][i]=i;
     for(i=1;i<=mm[n];i++)
     for(j=1;j<=n+1-(1<<i);j++)
     {
       a=best[i-1][j];
       b=best[i-1][j+(1<<(i-1))];
       if(RMQ[a]<RMQ[b]) best[i][j]=a;
       else best[i][j]=b;
     }
     return;
}
int askRMQ(int a,int b)
{
    int t;
    t=mm[b-a+1];b-=(1<<t)-1;
    a=best[t][a];b=best[t][b];
    return RMQ[a]<RMQ[b]?a:b;
}
int lcp(int a,int b)
{
    int t;
    a=Rank[a];b=Rank[b];
    if(a>b) {t=a;a=b;b=t;}
    return(height[askRMQ(a+1,b)]);
}

char str[maxn] , str2[maxn] ;
int sa[maxn] ;

int main(){
	while( scanf( "%s" , str ) != EOF ) {
		strcpy( str2 , str ) ;
		//strrev( str2 ) ;
		int len = strlen( str ) ;
		str[len] = '$' ; str[len+1] = 0 ;
		//strcat( str , str2 ) ;
		for( int i = 0 ; i < len ;i ++ ) {
			str[len+i+1] = str2[len-i-1] ;
		}
		str[len+len+1] = 0 ;
		len = strlen( str ) ;
		// 最后再加一个之前没有出现过的字符 ， 避免了判断边界情况
		str[len] = '&' ; str[len+1] = 0 ;
		len ++ ;
		da( str , sa , len + 1 , 200 ) ;
		calheight( str , sa , len ) ;
		initRMQ( len ) ;
		len -- ;
		int MaxLen = 0 , start = 0 ;
		for( int i = 0 ; i < len / 2 ; i ++ ) {
			// 长度为偶数的
			int tmp = lcp( i , len - i  ) * 2 ; 
			if( tmp > MaxLen ) {
				MaxLen = tmp ;
				start = i - tmp / 2 ;
			}
			// 长度是奇数的
			tmp = lcp( i , len - i - 1 ) * 2 - 1 ;
			if( tmp > MaxLen ) {
				MaxLen = tmp ;
				start = i - tmp / 2 ;
			}
		}
		while( MaxLen -- )
			putchar( str[start++] ) ;
		puts( "" ) ;
	}
	return 0 ;
}