URAL 1297 求最长回文字符串

有种简单的方法，数组从左到右扫一遍，每次以当前的点为中心，只要左右相等就往左右走，这算出来的回文字符串是奇数长度的

还有偶数长度的回文字符串就是以当前扫到的点和它左边的点作为中心，然后往左右扫

这是O(n^2)的复杂度，这道题过还是没有问题的

 

这里我主要练习的是另外的利用后缀数组加RMQ算法来解决这个问题

大致思想跟上面一致

首先将字符串反转贴在原字符串末尾，将字符通过ASCII码转化为字符，之间用一个1分开，最后贴一个0

然后得到它的后缀数组以及height[]数组

那么找回文字符也是扫一遍，每次以当前点作为偶数情况和奇数情况的中心

然后找到后来倒置的字符串对应字符的位置，这样二者的前缀一个是往前走的，一个是往后走的，正好贴一块就是回文的

但是这个查询为加快速度，就要用RMQ算法，求得区间内height数组的最小值

 

#include <cstdio>
#include <cstring>
#include <iostream>
#include <cmath>
using namespace std;
#define MAXN 2010
int sa[MAXN] , height[MAXN] , _rank[MAXN];
int wa[MAXN] , wb[MAXN] , wsf[MAXN] , wv[MAXN] , r[MAXN];
int dp[MAXN][15];
char str[MAXN];

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

void getSa(int *r , int *sa , int n , int m)
{
    int i,j,p;
    int *x=wa , *y=wb , *t;

    for(i=0 ; i<m ; i++) wsf[i]=0;
    for(i=0 ; i<n ; i++) wsf[x[i]=r[i]]++;
    for(i=1 ; i<m ; i++) wsf[i]+=wsf[i-1];
    for(i=n-1 ; i>=0 ; i--) sa[--wsf[x[i]]]=i;

    p=1;
    for(j=1 ; p<n ; j*=2 , 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<n ; i++) wv[i]=x[y[i]];
        for(i=0 ; i<m ; i++) wsf[i]=0;
        for(i=0 ; i<n ; i++) wsf[wv[i]]++;
        for(i=1 ; i<m ;  i++) wsf[i]+=wsf[i-1];
        for(i=n-1 ; i>=0 ; i--)sa[--wsf[wv[i]]]=y[i];

        t=x , x=y , y=t;
        x[sa[0]]=0;
        for(i=1,p=1;i<n;i++)
            x[sa[i]] = cmp(y , sa[i-1] , sa[i] , j)?p-1:p++;
    }
    return ;
}

void callHeight(int *r , int *sa , int n)
{
    int i,j,k=0;
    for(i=1 ; i<=n ; i++) _rank[sa[i]]=i;
  //  for(i=0 ; i<n ; i++) cout<<"_rank: "<<_rank[i]<<endl;
    for(i=0 ; i<n ; height[_rank[i++]]=k)
        for(k?k--:0 , j=sa[_rank[i]-1]; r[i+k]==r[j+k] ; k++);
    return ;
}


void ST(int n)
{
    //将height数组进行RMQ询问
    memset(dp , 0x3f , sizeof(dp));
    for(int i=1 ; i<=n ; i++) dp[i][0]=height[i];
    for(int j=1 ; (1<<(j-1))<n+1 ; j++)
        for(int i=1 ; i<=n ; i++)
            dp[i][j] = min(dp[i][j-1] , dp[i+(1<<(j-1))][j-1]);

    return;
}

int get_min(int s , int t)
{
    int len = t-s+1;
    int l=floor(log(len*1.0)/log(2.0));
    return min(dp[s][l] , dp[t-(1<<l)+1][l]);
}

void solve(int n , int &left , int &right)
{
    int ans = 0;
    //奇数情况
    for(int i=0 ; i<n ; i++){
        int j=2*n-i;
        int s=min(_rank[i] , _rank[j]);
        int t=max(_rank[i] , _rank[j]);
        int tmp = get_min(s+1 , t);
        if(ans<tmp*2-1){
            ans = tmp*2-1;
            left=i-tmp+1 , right=i+tmp-1;
        }
    }
    //偶数情况
    for(int i=0 ; i<n ; i++){
        int j = 2*n-i+1;
        int s=min(_rank[i] , _rank[j]);
        int t=max(_rank[i] , _rank[j]);
        int tmp = get_min(s+1 , t);
        if(ans<tmp*2){
            ans = tmp*2;
            left=i-tmp , right=i+tmp-1;
        }
    }
    return;
}

int main()
{
  //  freopen("a.in" , "r" , stdin);
    while(~scanf("%s" , str))
    {
        int len = strlen(str);
        r[len]=1;
        for(int i=0 ; i<len ; i++){
            r[i] = (int)str[i];
            r[2*len-i] = r[i];
        }
        r[2*len+1]=0;

        getSa(r , sa , 2*len+2 , 255);
        callHeight(r , sa , 2*len+1);
        //建立RMQ查询数组
        ST(2*len+1);
        int left=1 , right=1;
        solve(len , left , right);
        str[right+1]='\0';
        printf("%s\n" , str+left);
    }
    return 0;
}