bzoj3676 [Apio2014]回文串(manacher+SA+二分+lcp)

manacher板子戳这里
我们用manacher枚举回文串，用SA求这个串的出现次数。（利用h数组二分求最大包含区间[l…r]，则出现次数为r-l+1）类似这道题。不错的好题。。。

#include <cstdio>
#include <cstring>
#include <algorithm>
#define ll long long
#define N 300010
char s[N<<1];
int a[N],n,Log[N],p[N<<1];
ll ans=0;
int rank[N<<1],rank1[N],sa[N],h[N],tmp[N],count[N],st[N][20];
void SA(){
    int m=27;Log[0]=-1;
    for(int i=1;i<=n;++i) Log[i]=Log[i>>1]+1;
    for(int i=1;i<=m;++i) count[i]=0;
    for(int i=1;i<=n;++i) count[a[i]]=1;
    for(int i=1;i<=m;++i) count[i]+=count[i-1];
    for(int i=n;i>=1;--i) rank[i]=count[a[i]];
    int k=0;
    for(int p=1;k!=n;p<<=1,m=k){
        for(int i=1;i<=m;++i) count[i]=0;
        for(int i=1;i<=n;++i) count[rank[i+p]]++;
        for(int i=1;i<=m;++i) count[i]+=count[i-1];
        for(int i=n;i>=1;--i) tmp[count[rank[i+p]]--]=i;
        for(int i=1;i<=m;++i) count[i]=0;
        for(int i=1;i<=n;++i) count[rank[tmp[i]]]++;
        for(int i=1;i<=m;++i) count[i]+=count[i-1];
        for(int i=n;i>=1;--i) sa[count[rank[tmp[i]]]--]=tmp[i];
        memcpy(rank1,rank,sizeof(rank1));
        rank[sa[1]]=k=1;
        for(int i=2;i<=n;++i){
            if(rank1[sa[i]]!=rank1[sa[i-1]]||rank1[sa[i]+p]!=rank1[sa[i-1]+p]) ++k;
            rank[sa[i]]=k;
        }
    }k=0;
    for(int i=1;i<=n;++i){
        if(rank[i]==1){h[1]=0;continue;}
        if(i==1||h[rank[i-1]]<=1) k=0;
        if(k) --k;
        while(a[i+k]==a[sa[rank[i]-1]+k]) ++k;
        h[rank[i]]=k;
    }
    for(int i=1;i<=n;++i) st[i][0]=h[i];
    for(int i=1;i<=Log[n];++i)
        for(int j=1;j+(1<<i-1)<=n;++j)
            st[j][i]=std::min(st[j][i-1],st[j+(1<<i-1)][i-1]);
}
inline int lcp(int l,int r){
    ++l;int t=Log[r-l+1];
    return std::min(st[l][t],st[r-(1<<t)+1][t]);
}
inline ll calc(int x,int len){//x--回文开始位置,len--回文长度
    //二分求包含子串a[x...x+len-1]的最大区间 
    x=rank[x];int xl,xr;
    if(h[x]<len) xl=x;
    else{
        int l=1,r=x-1;
        while(l<=r){
            int mid=l+r>>1;
            if(lcp(mid,x)>=len) r=mid-1;else l=mid+1;
        }xl=l;
    }
    if(h[x+1]<len) xr=x;
    else{
        int l=x+1,r=n;
        while(l<=r){
            int mid=l+r>>1;
            if(lcp(x,mid)>=len) l=mid+1;else r=mid-1;
        }xr=r;
    }
    return (ll)len*(xr-xl+1);//出现次数为xr-xl+1 
}
int main(){
//  freopen("a.in","r",stdin);
    scanf("%s",s+1);n=strlen(s+1);
    for(int i=1;i<=n;++i) a[i]=s[i]-'a'+1;
    SA();int len=0;
    for(int i=1;i<=n;++i) s[++len]='#',s[++len]=a[i]+'a'-1;
    s[++len]='#';int id=0;
    for(int i=1;i<=len;++i){
        if(i<id+p[id]) p[i]=std::min(p[(id<<1)-i],id+p[id]-i);
        else p[i]=1;
        while(i-p[i]>=1&&i+p[i]<=len&&s[i-p[i]]==s[i+p[i]]){
            ++p[i];if(s[i-p[i]+1]!='#') continue;//回文一定以"#"结尾
            //if(i+p[i]<id+p[id]) 则这个回文一定出现过，不用再计算了 
            if(i+p[i]>id+p[id]) ans=std::max(ans,calc(i/2-p[i]/2+1,p[i]-1));
        }
        if(i+p[i]>id+p[id]) id=i;
    }
    printf("%lld\n",ans);
    return 0;
}