[BZOJ3238][Ahoi2013]差异

原题地址

题目大意：

题目的关键在于求$\Sigma lcp(T_i,T_j),(1≤i<j≤n).$
$用后缀数组求出hei数组,则lcp(T_i,T_j)对应于hei数组的rmq(i+1,j),然后题目转化为求\Sigma rmq(i,j),(2≤i≤j≤n).$
$然后每个区间hei[i],...,hei[j]一定有一个在最左边的最小值，枚举这个最小值，再二分区间统计即可.$

AC code:

#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;
typedef long long ll;
const ll N=500010;
const ll LOGN=20;
ll   l,ans;
ll   lg2[N],sa[N],t[N],c[N],rk[N],r1[N],r2[N],h[N],hei[N];
ll   f[N][LOGN];
char s[N];

void build(){
    for(ll i=0;i<l;i++) c[s[i]]=1;
    for(ll i=1;i<(1<<10);i++) c[i]+=c[i-1];
    for(ll i=0;i<l;i++) rk[i]=c[s[i]];
    for(ll i=1,j;rk[sa[l]]!=l;i<<=1){
        for(j=0;j<l;j++) r1[j]=j+i<l?rk[j+i]:0;
        for(j=0;j<l;j++) c[j]=0;
        for(j=0;j<l;j++) c[r1[j]]++;
        for(j=1;j<l;j++) c[j]+=c[j-1];
        for(j=0;j<l;j++) t[c[r1[j]]--]=j;
        for(j=0;j<l;j++) c[j]=0;
        for(j=0;j<l;j++) c[rk[j]]++;
        for(j=1;j<l;j++) c[j]+=c[j-1];
        for(j=l;j>0;j--) sa[c[rk[t[j]]]--]=t[j];
        for(r2[sa[1]]=1,j=2;j<=l;j++)
            r2[sa[j]]=r2[sa[j-1]]+(rk[sa[j]]!=rk[sa[j-1]]||r1[sa[j]]!=r1[sa[j-1]]);
        for(j=0;j<l;j++) rk[j]=r2[j];
    }
    for(ll i=0,j=0;i<l;h[i]=j,j=j?j-1:0,i++){
        for(;rk[i]!=1&&s[sa[rk[i]]+j]==s[sa[rk[i]-1]+j];j++) ;
    }
    for(ll i=0;i<l;i++) hei[rk[i]]=h[i];
    for(ll i=1;i<=l;i++) f[i][0]=hei[i];
    for(ll i=1,j;i<=lg2[l];i++){
        for(j=1;j<=l;j++){
            f[j][i]=f[j][i-1];
            if(j+(1<<(i-1))<=l) f[j][i]=min(f[j][i],f[j+(1<<(i-1))][i-1]);
        }
    }
}

ll rmq(ll f[N][LOGN],ll L,ll R){
    return min(f[L][lg2[R-L+1]],f[R-(1<<lg2[R-L+1])+1][lg2[R-L+1]]);
}

int main(){
    scanf("%s",s);
    l=strlen(s);
    for(ll i=1;i<=l;i++) lg2[i]=(ll)log2(i);
    for(ll i=l;i>=2;i--) ans+=(i*(i-1)*3)>>1;
    build();
    for(ll i=2;i<=l;i++){
        ll L1=1,R1=i,L2=i,R2=l+1;
        while(L1+1!=R1){
            ll M=(L1+R1)>>1;
            if(rmq(f,M,i-1)>hei[i]) R1=M;
            else L1=M;
        }
        while(L2+1!=R2){
            ll M=(L2+R2)>>1;
            if(rmq(f,i,M)<hei[i]) R2=M;
            else L2=M;
        }
        ans-=(hei[i]*(i-R1+1)*(L2-i+1))<<1;
    }
    printf("%lld\n",ans);

    return 0;
}