hdu 3068 最长回文子串 TLE

后缀数组+RMQ是O(nlogn)的，会TLE.....

标准解法好像是马拉车，O(n)....

#include "algorithm"
#include "cstdio"
#include "cstring"
using namespace std;
#define maxn 220020

int wa[maxn],wb[maxn],wv[maxn],ws[maxn];
int rank[maxn],height[maxn];
int r[maxn],sa[maxn],RMQ[maxn][20];
char s[110010];
int n;

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 *sa,int n,int m)
{
    int i,j,p,*x=wa,*y=wb,*t;
    for(i=0; i<m; i++) ws[i]=0;     //注意此处有个小问题：ws和iostream里面某地重名了..
    for(i=0; i<n; i++) ws[x[i]=r[i]]++;
    for(i=1; i<m; i++) ws[i]+=ws[i-1];
    for(i=n-1; i>=0; i--) sa[--ws[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++) ws[i]=0;
        for(i=0; i<n; i++) ws[wv[i]]++;
        for(i=1; i<m; i++) ws[i]+=ws[i-1];
        for(i=n-1; i>=0; i--) sa[--ws[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++;
    }
    return;
}

void calheight(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; height[rank[i++]]=k)
        for(k?k--:0,j=sa[rank[i]-1]; r[i+k]==r[j+k]; k++);
    return;
}

bool _same(int lx,int ly,int l1,int l2)
{
    return (((lx<=l1-1) && (ly>=l1+1))||((ly<=l1-1) && (lx>=l1+1)));
}

void ST()        //初始化
{
    memset(RMQ,1,sizeof(RMQ));
    for(int i=1;i<=n;i++)
        RMQ[i][0]=height[i];
    for(int j=1;(1<<j)<=n;j++)
        for(int i=1;i+(1<<j)-1<=n;i++)
        RMQ[i][j]=min(RMQ[i][j-1],RMQ[i+(1<<(j-1))][j-1]);
}

int Query(int L,int R)    //求a[L..R]区间的最值
{
    int k=0;
    while((1<<(k+1))<=R-L+1) k++;
    int tb=min(RMQ[L][k],RMQ[R-(1<<k)+1][k]);
    return tb;
}

int calc(int l,int r)       //求l开始后缀和r开始后缀的最长公共前缀
{
    int tl=rank[l],tr=rank[r];
    if (tl>tr)      swap(tl,tr);
    tl++;
    int ans=Query(tl,tr);       //相当于RMQ问题
//  printf("calc:  %d %d -- %d %d == %d\n",l,r,tl,tr,ans);
    return ans;
}

int _max(int a,int b,int c,int d)
{
    int mx=a;
    if (b>mx) mx=b;
    if (c>mx) mx=c;
    if (d>mx) mx=d;
    return mx;
}

//da(r,sa,n+1,128);
//calheight(r,sa,n);
int main()
{
    while(~scanf("%s",s))
    {
        n=strlen(s);

        for (int i=0; i<n; i++)
            r[i]=int(s[i])-int('a')+1;

        r[n]=100;
        for (int i=n-1; i>=0; i--)
            r[2*n-i]=int(s[i])-int('a')+1;
        r[2*n+1]=0;
        int n2=n;       n=2*n+1;

//      for (int i=0; i<=n; i++)      printf("%d ",r[i]);
//        printf("\n %d   %d\n",n,n2);

        da(r,sa,n+1,128);
        calheight(r,sa,n);

//      for (int i=0; i<=n; i++)
//          printf("%d  %d  %d\n",sa[i],height[i],rank[i]);
//      printf("\n");

        ST();
        int ans=1;
        for (int i=1; i<=n2-2; i++)
        {
            int t1=calc(i,n-i)*2;
            int t2=calc(i+1,n-i)*2+1;
            int t3=calc(i+1,n-i-1)*2;
//          printf("%d %d %d\n",t1,t2,t3);
            ans=_max(ans,t1,t2,t3);
        }
        printf("%d\n",ans);
    }
    return 0;
}

 

 

 

 

 

 

 

 

 

最近两天有点不在状态....先滚去整理模板吧