字符串(马拉车算法，后缀数组，稀疏表)：BZOJ 3676 [Apio2014]回文串

  1 #include <iostream>
  2 #include <cstring>
  3 #include <cstdio>
  4 using namespace std;
  5 const int maxn=300100;
  6 char s[maxn],S[maxn<<1];
  7 int maxl[maxn<<1];
  8 int pos[maxn<<1],maxL[maxn];
  9 int rank[maxn],Wa[maxn],Wb[maxn],Wv[maxn],Ws[maxn],sa[maxn],r[maxn];
 10 int lcp[maxn],len;
 11 int mm[maxn],Min[maxn][32];
 12 bool cmp(int *p,int a,int b,int l){
 13     return p[a]==p[b]&&p[a+l]==p[b+l];
 14 }
 15 
 16 void DA(int n,int m){
 17     int i,j,p,*x=Wa,*y=Wb,*t;
 18     for(i=0;i<m;i++)Ws[i]=0;
 19     for(i=0;i<n;i++)++Ws[x[i]=r[i]];
 20     for(i=1;i<m;i++)Ws[i]+=Ws[i-1];
 21     for(i=n-1;i>=0;i--)sa[--Ws[x[i]]]=i;
 22     
 23     for(j=1,p=1;p<n;j<<=1,m=p){
 24         for(p=0,i=n-j;i<n;i++)y[p++]=i;
 25         for(i=0;i<n;i++)
 26             if(sa[i]>=j)
 27                 y[p++]=sa[i]-j;
 28         
 29         for(i=0;i<m;i++)Ws[i]=0;
 30         for(i=0;i<n;i++)++Ws[Wv[i]=x[y[i]]];
 31         for(i=1;i<m;i++)Ws[i]+=Ws[i-1];
 32         for(i=n-1;i>=0;i--)sa[--Ws[Wv[i]]]=y[i];
 33         for(t=x,x=y,y=t,x[sa[0]]=0,i=1,p=1;i<n;i++)    
 34             x[sa[i]]=cmp(y,sa[i],sa[i-1],j)?p-1:p++;
 35     }
 36 }
 37 
 38 void Lcp(int n){
 39     int i,j,k=0;
 40     for(i=1;i<=n;i++)rank[sa[i]]=i;
 41     for(i=0;i<n;lcp[rank[i++]]=k)
 42         for(k?--k:k,j=sa[rank[i]-1];r[j+k]==r[i+k];++k);
 43 }
 44 int Qmin(int l,int r){
 45     return min(Min[l][mm[r-l+1]],Min[r-(1<<mm[r-l+1])+1][mm[r-l+1]]);
 46 }
 47 
 48 int Query(int p,int l){
 49     int ret=1,lo,hi;
 50     hi=rank[p];lo=0;
 51     while(lo<=hi){
 52         int mid=(lo+hi)>>1;
 53         if(Qmin(mid,rank[p])>=l)hi=mid-1;
 54         else lo=mid+1;
 55     }
 56     ret+=rank[p]-lo+1;    
 57     
 58     hi=len;lo=rank[p]+1;
 59     while(lo<=hi){
 60         int mid=(lo+hi)>>1;
 61         if(Qmin(rank[p]+1,mid)>=l)lo=mid+1;
 62         else hi=mid-1;
 63     }
 64     return ret+hi-rank[p];
 65 }
 66 
 67 int main(){
 68     scanf("%s",s);
 69     len=strlen(s);
 70     int l=0;
 71     S[l++]=‘#‘;S[l++]=‘&‘;
 72     for(int i=0;i<len;i++){
 73         pos[l]=i;
 74         S[l++]=s[i];
 75         pos[l]=i+1;
 76         S[l++]=‘&‘;
 77     }
 78     S[l]=0;
 79     int mx=0,id;
 80     for(int i=1;i<l;i++){
 81         maxl[i]=mx>i?min(maxl[id*2-i],mx-i):1;
 82         while(S[i+maxl[i]]==S[i-maxl[i]])maxl[i]++;
 83         if(i+maxl[i]>mx){
 84             mx=i+maxl[i];
 85             id=i;
 86         }
 87     }
 88     
 89     for(int i=1;i<=l;i++)
 90         maxL[pos[i-maxl[i]+1]]=max(maxL[pos[i-maxl[i]+1]],maxl[i]-1);
 91     //maxL[i]原s字符串中i位置最长回文子串
 92     for(int i=0;i<len;i++)r[i]=s[i];
 93     DA(len+1,128);
 94     Lcp(len);
 95     
 96     mm[0]=-1;
 97     for(int i=1;i<=len;i++){
 98         mm[i]=(i&(i-1))?mm[i-1]:mm[i-1]+1;
 99         Min[i][0]=lcp[i];
100     }
101     
102     for(int k=1;k<=mm[len];k++)
103     for(int i=1;i+(1<<k)-1<=len;i++)
104     Min[i][k]=min(Min[i][k-1],Min[i+(1<<(k-1))][k-1]);
105     
106     long long ans=0;
107     for(int i=0;i<len;i++){
108         ans=max(ans,1ll*maxL[i]*Query(i,maxL[i]));
109     }
110     printf("%lld\n",ans);
111     return 0;
112 }