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;}