ural 1297 Palindrome 后缀数组

      给一个长度不超过1000的由大小写字母组成的字符串，求最长回文子串。一眼看上去就是一道裸题竟然调了这么久...一开始的思路是把原串翻转接到原串后面（用分隔符隔开，并且串为加分隔符）构造后缀数组之后直接枚举相邻的sa，判断两个位置是否在不同的串里并且对应同一个位置。后来各种WA。给个数据aacdlkoiijtfidcaa ，因为只枚举相邻后缀的话，aa这个最早的回文串根本找不出来...后来的做法是求出后缀数组后，先循环一边，用一个数组（lock）记录一下原串中第i位和翻转后接在后面的len-i+len位在sa中对应的位置，然后就是枚举回文串的中间字符了,枚举到原串第i位时，lock[i],lock[len-i+len]两个后缀的最长公共前缀就是中心两侧相同字符的长度。枚举的时候，对回文串的奇偶也要分情况求。

#include <iostream>
#include <algorithm>
#include <cmath>
#include <cstdio>
#include <algorithm>
#include <stack>
#include <queue>
#include <map>
#include <string>
#include <cstring>
#include <string>
using namespace std;
typedef long long ll;
const int maxn=40000+40;
int s[maxn],rs[maxn];
int sa[maxn],t[maxn],t2[maxn],c[maxn];
int n,m,k,tt;
char s1[maxn],s2[maxn];
int rank[maxn],height[maxn];
int l1,l2;

inline int idx(char s)
{
    if (s>='a' && s<='z') return 2+s-'a';
    return 2+s-'A'+26;
}
void getheight(int n)
{
    int i,j,k=0;
    for (i=0; i<=n; i++) rank[sa[i]]=i;
    for (i=0; i<n; i++)
    {
        if (k) k--;
        int j=sa[rank[i]-1];
        while(s[i+k]==s[j+k]) k++;
        height[rank[i]]=k;
    }
}

void build_ss(int m,int n)
{
    n++;
    int i,*x=t,*y=t2;
    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 (i=n-k; i<n; i++) y[p++]=i;
        for (i=0; i<n; i++) if (sa[i]>=k) y[p++]=sa[i]-k;

        for (i=0; i<m; i++) c[i]=0;
        for (i=0; i<n; i++) c[x[y[i]]]++;
        for (i=1; i<m; i++) c[i]+=c[i-1];
        for (i=n-1; i>=0; i--) sa[--c[x[y[i]]]] = y[i];
        swap(x,y);
        p=1;
        x[sa[0]]=0;
        for (i=1; i<n; i++)
        x[sa[i]]=(y[sa[i-1]]==y[sa[i]] && y[sa[i-1]+k]==y[sa[i]+k])? p-1 : p++;
        if (p>=n) break;
        m=p;
    }
}

bool diff(int x,int y,int len)
{
//    if (x<len && y>len && (y+hg-1==len-x+len)) return true;
//    if (x>len && y<len && (x+hg-1==len-y+len)) return true;
//    return false;
    if (x<len && y>len) return true;
    return false;
}
int ans;
void print(int x,int len)
{
    for (int i=x; i<len; i++)
    cout<<char(s[i]-2+'a');
    cout<<endl;
}
int d[maxn][20];
int RMQ_init(int* a,int n)
{
    for (int i=1; i<=n; i++)
    d[i][0]=a[i];
    for (int j=1; (1<<j)<=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 RMQ(int L,int R)
{
    if (L>R) swap(L,R);
    L++;
    int k=0;
    while ((1<<(k+1))<=R-L+1) k++;
    return min(d[L][k],d[R-(1<<k)+1][k]);
}

int lock[maxn];
int main()
{
//    freopen("in.txt","r",stdin);
//    freopen("out.txt","w",stdout);
    while(~scanf("%s",s1))
    {
        memset(c,0,sizeof c);
        memset(sa,0,sizeof sa);
        int len=strlen(s1);
        m=0;
        for (int i=0; i<len; i++)
        {
            s[i]=idx(s1[i]);
            m=max(m,s[i]);
        }
        s[len]=1;
        k=len+1;
        for (int i=len-1; i>=0; i--)
        {
            s[k++]=idx(s1[i]);
        }
        s[k]=0;
        build_ss(m+1,k);
        getheight(k);
        for (int i=1; i<=k; i++)
        {
            lock[sa[i]]=i;
        }
        RMQ_init(height,k);
        ans=0;
        int tmp,tmq,tgt;
        for (int i=0; i<len; i++)
        {
            tmp=RMQ(lock[i],lock[len+len-i]);
            tmq=(tmp-1)<<1|1;
            if (ans<tmq)
            {
                ans=tmq;
                tgt=i-tmp+1;
            }
            else if (ans==tmq)
            {
                tgt=min(tgt,i-tmp+1);
            }

            if (i>0)
            {
                tmp=RMQ(lock[i],lock[len+len-(i-1)]);
                tmq=tmp<<1;
                if (ans<tmq)
                {
                    ans=tmq;
                    tgt=i-tmp;
                }
                else if (ans==tmq)
                {
                    tgt=min(tgt,i-tmp);
                }

            }
        }
        int i,j;
        for (i=tgt,j=0;j<ans;i++,j++)
        printf("%c",s1[i]);
        printf("\n");

//        for (int i=1; i<=k; i++) cout<<i<<" ",print(sa[i],k);

    }
    return 0;
}