http://fastvj.rainng.com/problem/URAL-1297
题意为求最长回文串。
这个题也好做,我们就把输入的字符串str和他自己反向链接,然后求sa数组和height数组;
最后遍历后缀数组,这个时候就要细心了,首先你要判断sa[i-1],sa[i]是不是分别属于两个字符串了,并且还有判断是不是回文串的起点,记录位置的变量初始化为0,就可以了
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1000010;
int s[N],x[N],y[N],n,m;
int c[N],sa[N],height[N],rk[N];
void Suffix()
{
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]]&&y[sa[i]+k]==y[sa[i-1]+k]?p-1:p++;
if(p>=n) break;
m=p;
}
}
void geth()
{
for(int i=0;i<n;i++) rk[sa[i]]=i;
for(int i=0,k=0;i<n;i++)
{
if(rk[i])
{
if(k) --k;
int j=sa[rk[i]-1];
while(s[i+k]==s[j+k])
k++;
height[rk[i]]=k;
}
}
}
int k;
char str1[N],str2[N];
int main()
{
while(~scanf("%s",str1))
{
n=0;
int l=strlen(str1);
for(int i=0;i<l;i++) s[n++]=str1[i]-'A'+1;
s[n++]=127;
for(int i=l-1;i>=0;i--) s[n++]=str1[i]-'A'+1;
s[n++]=0;
m=128;
Suffix();
geth();
int ans=1,pos=0;
for(int i=1;i<n;i++)
{
int mx=max(sa[i],sa[i-1]);
int mn=min(sa[i],sa[i-1]);
if(mx<l||mn>l) continue;
if(mn+height[i]!=n-1-mx) continue;
if(height[i]>ans)
{
ans=height[i];
pos=mn;
}
if(height[i]==ans) pos=min(mn,pos);
}
for(int i=pos;i<pos+ans;i++) printf("%c",s[i]+'A'-1);
printf("\n");
}
return 0;
}