URAL 1297 后缀数组+线段树

//By SiriusRen
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define N 5555
int cases,pos,maxans,n,cntA
,cntB
,A
,B
,rk
,sa
,tsa
,ht
;
char s
;
void SA(){
memset(cntA,0,sizeof(cntA));
for(int i=1;i<=n;i++)cntA[s[i]]++;
for(int i=1;i<=256;i++)cntA[i]+=cntA[i-1];
for(int i=n;i;i--)sa[cntA[s[i]]--]=i;
rk[sa[1]]=1;
for(int i=2;i<=n;i++)rk[sa[i]]=rk[sa[i-1]]+(s[sa[i]]!=s[sa[i-1]]);
for(int l=1;rk[sa
]<n;l<<=1){
memset(cntA,0,sizeof(cntA));
memset(cntB,0,sizeof(cntB));
for(int i=1;i<=n;i++)
cntA[A[i]=rk[i]]++,
cntB[B[i]=i+l<=n?rk[i+l]:0]++;
for(int i=1;i<=n;i++)cntA[i]+=cntA[i-1],cntB[i]+=cntB[i-1];
for(int i=n;i;i--)tsa[cntB[B[i]]--]=i;
for(int i=n;i;i--)sa[cntA[A[tsa[i]]]--]=tsa[i];
rk[sa[1]]=1;
for(int i=2;i<=n;i++)rk[sa[i]]=rk[sa[i-1]]+(A[sa[i]]!=A[sa[i-1]]||B[sa[i]]!=B[sa[i-1]]);
}
for(int i=1,j=0;i<=n;i++){
j=j?j-1:0;
while(s[i+j]==s[sa[rk[i]-1]+j])j++;
ht[rk[i]]=j;
}
}
int tree[N*8];
void build(int l,int r,int pos){
if(l==r){tree[pos]=ht[l];return;}
int mid=(l+r)>>1,lson=pos<<1,rson=pos<<1|1;
build(l,mid,lson),build(mid+1,r,rson);
tree[pos]=min(tree[lson],tree[rson]);
}
int query(int l,int r,int pos,int L,int R){
if(l>=L&&r<=R)return tree[pos];
int mid=(l+r)>>1,lson=pos<<1,rson=pos<<1|1;
if(mid<L)return query(mid+1,r,rson,L,R);
else if(mid>=R)return query(l,mid,lson,L,R);
else return min(query(l,mid,lson,L,R),query(mid+1,r,rson,L,R));
}
int main(){
scanf("%s",s+1),n=strlen(s+1);
s[n+1]='#';
for(int i=1;i<=n;i++)s[n+i+1]=s[n-i+1];
n=n+n+1;
SA(),build(1,n,1);
for(int i=1;i<=n/2;i++){
int tempx=min(rk[i],rk[n-i+1]),tempy=max(rk[i],rk[n-i+1]),jy;
jy=query(1,n,1,tempx+1,tempy);
if(maxans<jy*2-1)maxans=jy*2-1,pos=i-jy+1;
else if(maxans==jy*2-1&&pos>i-jy+1)pos=i-jy+1;
tempx=min(rk[i+1],rk[n-i+1]),tempy=max(rk[i+1],rk[n-i+1]);
jy=query(1,n,1,tempx+1,tempy);
if(maxans<jy*2)maxans=jy*2,pos=i-jy+1;
else if(maxans==jy*2&&pos>i-jy+1)pos=i-jy+1;
}
for(int i=pos;i<maxans+pos;i++)printf("%c",s[i]);
}