URAL 1297 Palindrome <后缀数组+RMQ>

题目：传送门

分析：求最长回文子串。将原串倒置链接在后面，中间用一个不会出现的字符隔开（这样任何两个后缀的LCP都不会跨过它），再求height数组。

代码：

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cstring>
using namespace std;

const int MAXN=2010;
int sa[MAXN],wm[MAXN],wa[MAXN],wb[MAXN],height[MAXN];
int *rank;
int dp[MAXN][12];

bool cmp(int* r,int a,int b,int len){
	return r[a]==r[b]&&r[a+len]==r[b+len];
}

void da(char *data,int n,int m){
	int *x=wa,*y=wb,*t,i,j,p;
	for(i=0;i<m;++i) wm[i]=0;
	for(i=0;i<n;++i) ++wm[x[i]=data[i]];
	for(i=1;i<m;++i) wm[i]+=wm[i-1];
	for(i=n-1;i>=0;--i) sa[--wm[x[i]]]=i;
	for(j=1,p=1;p<n;j<<=1,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<m;++i) wm[i]=0;
		for(i=0;i<n;++i) ++wm[x[y[i]]];
		for(i=1;i<m;++i) wm[i]+=wm[i-1];
		for(i=n-1;i>=0;--i) sa[--wm[x[y[i]]]]=y[i];
		for(t=x,x=y,y=t,i=p=1,x[sa[0]]=0;i<n;++i) x[sa[i]]=cmp(y,sa[i],sa[i-1],j)?p-1:p++;
	}
	rank=x; 
}

void getHeight(char *data,int n){
	for(int i=0,j,k=0;i<n;height[rank[i++]]=k)
		for(k?--k:k,j=sa[rank[i]-1];data[i+k]==data[j+k];++k);
}

void initRMQ(int n){
	int l=(int)(log(n*1.0)/log(2.0));
	for(int i=1;i<=n;++i) dp[i][0]=height[i];
	for(int j=1;j<=l;++j){
		for(int i=1;i+(1<<j)-1<=n;++i){
			dp[i][j]=min(dp[i][j-1],dp[i+(1<<(j-1))][j-1]);
		}
	}
}

int LCP(int l,int r){
	int a=rank[l],b=rank[r];
	if(a>b) swap(a,b); ++a;
	int j=(int)(log(1.0*(b-a+1))/log(2.0));
	return min(dp[a][j],dp[b-(1<<j)+1][j]);
}

void solve(char *str,int n){
	int max_len=1,star=0;
	for(int i=0;i<n;++i){
		int len=2*LCP(i,n*2-i)-1;
		if(len>max_len) max_len=len,star=i-len/2;
		if(i){
			len=2*LCP(i,n*2-i+1);
			if(len>max_len) max_len=len,star=i-len/2;
		}
	}
	str[star+max_len]='\0';
	printf("%s\n",&str[star]);
}

int main(){
	char str[MAXN];
	scanf("%s",str);
	int n=strlen(str); str[n]='#';
	for(int i=n-1;i>=0;--i) str[n*2-i]=str[i];
	str[2*n+1]=0;
	da(str,2*n+2,260);
	getHeight(str,n*2+1);
	initRMQ(n*2+1);
	solve(str,n);
	return 0;
}