bzoj 3676 [Apio2014]回文串（Manacher+SAM）

  1 #include<set>
  2 #include<cmath>
  3 #include<queue>
  4 #include<vector>
  5 #include<cstdio>
  6 #include<cstring>
  7 #include<iostream>
  8 #include<algorithm>
  9 #define trav(u,i) for(int i=front[u];i;i=e[i].nxt)
 10 #define FOR(a,b,c) for(int a=(b);a<=(c);a++)
 11 #define rep(a,b,c) for(int a=(b);a>=(c);a--)
 12 using namespace std;
 13 
 14 typedef long long ll;
 15 const int N = 6e5+10;
 16 const int D = 21;
 17 
 18 char s[N];
 19 int n,p[N];
 20 
 21 struct SAM 
 22 {
 23     
 24     ll ans;
 25     int sz,last,ch[N][26],fa[N],R[N],pos[N],l[N],b[N],cnt[N],fat[N][D];
 26     SAM() 
 27     {
 28         sz=ans=0; last=++sz;
 29         memset(cnt,0,sizeof(cnt));
 30         memset(R,0,sizeof(R));
 31         memset(fat,0,sizeof(fat));
 32     }
 33     void add(int c,int id)
 34     {
 35         int np=++sz,p=last; last=np;
 36         l[np]=l[p]+1; R[np]=1; pos[id]=last;
 37         for(;p&&!ch[p][c];p=fa[p]) ch[p][c]=np;
 38         if(!p) fa[np]=1;
 39         else {
 40             int q=ch[p][c];
 41             if(l[q]==l[p]+1) fa[np]=q;
 42             else {
 43                 int nq=++sz; l[nq]=l[p]+1;
 44                 memcpy(ch[nq],ch[q],sizeof(ch[q]));
 45                 fa[nq]=fa[q];
 46                 fa[q]=fa[np]=nq;
 47                 for(;ch[p][c]==q;p=fa[p]) ch[p][c]=nq; 
 48             }
 49         }
 50     }
 51     void get_pre()
 52     {
 53         FOR(i,1,sz) cnt[l[i]]++;
 54         FOR(i,1,n) cnt[i]+=cnt[i-1];
 55         rep(i,sz,1) b[cnt[l[i]]--]=i;
 56         rep(i,sz,1) R[fa[b[i]]]+=R[b[i]];
 57         
 58         FOR(i,1,sz) {
 59             fat[i][0]=fa[i];
 60             FOR(j,1,D-1)
 61                 fat[i][j]=fat[fat[i][j-1]][j-1];
 62         }
 63     }
 64     void get_ans(int u,int v)
 65     {
 66         int x=pos[v];
 67         for(int i=D-1;i>=0;i--) {
 68             int t=fat[x][i];
 69             if(l[t]>=v-u+1) x=t;
 70         }
 71         ans=max(ans,(ll)R[x]*(v-u+1));
 72     }
 73     
 74 } sam;
 75 
 76 void Manacher()
 77 {
 78     int mx=0,id;
 79     for(int i=1;i<=n;i++) {
 80         if(mx>i) p[i]=min(mx-i,p[2*id-i-1]);
 81         else p[i]=0;
 82         while(s[i+p[i]+1]==s[i-p[i]]) {
 83             p[i]++;
 84             sam.get_ans(i-p[i]+1,i+p[i]);
 85         }
 86         if(p[i]+i>mx) mx=p[i]+i,id=i;
 87     }
 88     mx=0;
 89     for(int i=1;i<=n;i++) {
 90         if(mx>i) p[i]=min(mx-i-1,p[2*id-i]);
 91         else p[i]=1,sam.get_ans(i,i);
 92         while(s[i+p[i]]==s[i-p[i]]) {
 93             p[i]++;
 94             sam.get_ans(i-p[i]+1,i+p[i]-1);
 95         }
 96         if(p[i]+i>mx) mx=p[i]+i,id=i;
 97     }
 98 }
 99 
100 int main()
101 {
102     scanf("%s",s+1);
103     n=strlen(s+1);
104     FOR(i,1,n) sam.add(s[i]-‘a‘,i);
105     sam.get_pre();
106     s[0]=‘+‘,s[n+1]=‘-‘;
107     Manacher();
108     printf("%lld",sam.ans);
109     return 0;
110 }