URAL 1297 后缀数组+线段树

思路:
论文题……*n
倒过来接上 分奇偶讨论 求LCP 搞棵线段树即可

//By SiriusRen
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define N 5555
int cases,pos,maxans,n,cntA[N],cntB[N],A[N],B[N],rk[N],sa[N],tsa[N],ht[N];
char s[N];
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]]<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]);
}

这里写图片描述

posted @ 2017-01-15 23:27  SiriusRen  阅读(194)  评论(0编辑  收藏  举报