题目:
题意:
求一个字符串中的最长回文,如果有多个长度相等的最长回文,那就输出第一个最长回文。
题解:
这是后缀数组的一种常见的应用,首先把原始字符串倒转过来,然后接在原始字符串的后面,中间用一个不可能出现的字符隔开。
我们枚举每一个原始字符串中的字符以它为中心(分为奇数和偶数两种情况)进行查找
- 奇数时,我们先把这一位算上去,然后再-1,查询范围为[i,n-i-1]
- 偶数时,查询范围为[i,n-i],相当于选择了一个奇数中心但并不可能匹配上ta
这样就把问题变为 了求这个新的字符串的某两个后缀的最长公共前缀
suffix(j) 和 suffix(k) 的最长公共前缀为 height[rank[j]+1],height[rank[j]+2], height[rank[j]+3], … ,height[rank[k]] 中的最小值,用RMQ求区间最值了
代码:
#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;
const int N=3005;
int c[N],x[N],y[N],m,n,height[N],rak[N],sa[N],f[N][25];
char s[N];
void build_sa()
{
m=300;
for (int i=0;i<m;i++) c[i]=0;
for (int i=0;i<n;i++) c[x[i]=s[i]]++;
for (int i=1;i<m;i++) c[i]+=c[i-1];
for (int i=n-1;i>=0;i--) sa[--c[x[i]]]=i;
for (int k=1;k<=n;k<<=1)
{
int p=0;
for (int i=n-k;i<n;i++) y[p++]=i;
for (int i=0;i<n;i++) if (sa[i]>=k) y[p++]=sa[i]-k;
for (int i=0;i<m;i++) c[i]=0;
for (int i=0;i<n;i++) c[x[y[i]]]++;
for (int i=1;i<m;i++) c[i]+=c[i-1];
for (int i=n-1;i>=0;i--) sa[--c[x[y[i]]]]=y[i];
swap(x,y);p=1;x[sa[0]]=0;
for (int i=1;i<n;i++) x[sa[i]]=y[sa[i]]==y[sa[i-1]] && ((sa[i]+k>=n?-1:y[sa[i]+k])==(sa[i-1]+k>=n?-1:y[sa[i-1]+k]))?p-1:p++;
if (p>n) break;
m=p;
}
}
void build_lcp()
{
for (int i=0;i<n;i++) rak[sa[i]]=i;
height[0]=0;int k=0;
for (int i=0;i<n;i++)
{
if (!rak[i]) continue;
if (k) k--;
int j=sa[rak[i]-1];
while (i+k<n && j+k<n && s[i+k]==s[j+k]) k++;
height[rak[i]]=k;
}
}
void rmq()
{
for (int i=0;i<n;i++) f[i][0]=height[i];
for (int i=1;i<20;i++)
for (int j=0;j<n;j++)
if (j+(1<<i)-1<n)
f[j][i]=min(f[j][i-1],f[j+(1<<(i-1))][i-1]);
else break;
}
int lcp(int a,int b)
{
int l=rak[a],r=rak[b];
if(r<l) swap(l,r);
if(r<0) return 0;l++;
int k=log2(r-l+1);
return min(f[l][k],f[r-(1<<k)+1][k]);
}
int main()
{
scanf("%s",s);
n=strlen(s);
s[n]='&';int nn=n,ans=1,wz=0,tmp;
for (int i=nn-1;i>=0;i--) s[++n]=s[i];
n++;
build_sa();
build_lcp();
rmq();
for(int i=0;i<nn;i++)
{
tmp=lcp(i,n-i-1);
if(2*tmp-1>ans) //对称个数为奇数
{
ans=2*tmp-1;
wz=i-tmp+1;
}
tmp=lcp(i,n-i);
if(2*tmp>ans) //对称个数为偶数
{
ans=2*tmp;
wz=i-tmp;
}
}
for (int i=wz;ans--;i++) printf("%c",s[i]);
}