UVAL 1297 Palindrome 最长回文子串

题意：给出一个字符串，求出最长回文子串。

思路：我们枚举中心，对于回文中心，我们可以看到以回文中心为前缀和后缀是相同的，这样我们就可以反转字符串（将前缀变成后缀），将两个字符串拼接起来，对于不同的回文中心，我们考察它的前缀和后缀的最长公共子串。这样，我们就能得到最大的回文子串。

注意：回文串需要分奇偶讨论。

代码如下：

#include <cstring>
#include <cstdio>
#include <algorithm>

using namespace std;


    static const int maxn =20010;
    int sa[maxn], rk[maxn], height[maxn];
    int wa[maxn], wb[maxn], wv[maxn], wd[maxn];

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

    void da(int *r, int n, int m){          //  倍增算法 r为待匹配数组  n为总长度 m为字符范围
        int i, j, p, *x = wa, *y = wb, *t;
        for(i = 0; i < m; i ++) wd[i] = 0;
        for(i = 0; i < n; i ++) wd[x[i]=r[i]] ++;
        for(i = 1; i < m; i ++) wd[i] += wd[i-1];
        for(i = n-1; i >= 0; i --) sa[-- wd[x[i]]] = i;
        for(j = 1, p = 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 ++) wd[i] = 0;
            for(i = 0; i < n; i ++) wd[wv[i]] ++;
            for(i = 1; i < m; i ++) wd[i] += wd[i-1];
            for(i = n-1; i >= 0; i --) sa[-- wd[wv[i]]] = y[i];
            for(t = x, x = y, y = t, p = 1, x[sa[0]] = 0, i = 1; i < n; i ++){
                x[sa[i]] = cmp(y, sa[i-1], sa[i], j) ? p - 1: p ++;
            }
        }
    }

    void calc_height(int *r, int n){           //  求height数组。
        int i, j, k = 0;
        for(i = 1; i <= n; i ++) rk[sa[i]] = i;
        for(i = 0; i < n; height[rk[i ++]] = k){
            for(k ? k -- : 0, j = sa[rk[i]-1]; r[i+k] == r[j+k]; k ++);
        }
    }
/**************************************/
    static const int MAX = 20010;
    int p[MAX];
    int d[MAX][20];

    void rmq_init(int n){
        p[0] = -1;
        for(int i = 1; i <= n; ++i)
            p[i] = i & (i-1)?p[i-1]:p[i-1]+1;
        for(int i = 1; i <= n; ++i) d[i][0] = height[i];
        for(int j = 1; j <= p[n]; ++j)
            for(int i = 1; i + (1 << j) - 1 <= n; ++i)
                d[i][j] = min(d[i][j-1],d[i+(1<<j-1)][j-1]);
    }

    int rmp_query(int l, int r){
        int k = p[r - l + 1];
        return min(d[l][k],d[r - (1<<k) + 1][k]);
    }

    int lcp(int l, int r){
        l = rk[l], r = rk[r];
        if(l > r) swap(l,r);l++;
        return rmp_query(l,r);
    }

/************************************************/
    void calc(int*r, int n, int m){
        r[n] = 0;
        da(r,n+1,m);
        calc_height(r,n);
        rmq_init(n);
    }

char str[MAX];
int r[MAX];

int main(void)
{
    //freopen("input.txt","r",stdin);
    int cas = 1;
    {
        scanf("%s",str);
        int n = strlen(str);
        for(int i = 0; i < n; ++i)
            r[i] = r[2 * n - i] = str[i];
        r[n] = '$';
        calc(r,2 * n + 1,256);
        int ans = 0;
        int pos = 0;
        for(int i = 0; i < n; ++i){
            int l = lcp(i,2 * n - i);
            if(2 * l - 1 > ans){
                ans = 2 * l - 1;
                pos = i - l + 1;
            }
        }
        for(int i = 1; i < n; ++i){
            int l = lcp(i,2 * n - i + 1);
            if(2 * l > ans){
                ans = 2 * l;
                pos = i - l;
            }
        }
        str[pos+ans] = 0;
        printf("%s",str+pos);
    }
    return 0;
}