AHOI2013 差异 【后缀数组】

题目分析：

求出height以后很明显跨越最小height的一定贡献是最小height，所以对于区间找出最小height再将区间对半分。

代码：

#include<bits/stdc++.h>
using namespace std;

const int maxn = 502000;
const int N = 500000;

int n;
char str[maxn];

int sa[maxn],rk[maxn],X[maxn],Y[maxn];
int height[maxn],h[maxn],RMQ[maxn][20],pos[maxn][20];

int chk(int x,int k){
    return rk[sa[x]]==rk[sa[x-1]]&&rk[sa[x]+(1<<k)]==rk[sa[x-1]+(1<<k)];
}

void getsa(){
    for(int i=0;i<n;i++) X[str[i]]++;
    for(int i=1;i<=N;i++) X[i] += X[i-1];
    for(int i=n-1;i>=0;i--) sa[X[str[i]]--] = i;
    for(int i = 2, num = 1;i <= n;i++)
    rk[sa[i]] = (str[sa[i]] == str[sa[i-1]]?num:++num);
    rk[sa[1]] = 1;
    for(int k=1;(1<<k-1)<=n;k++){
    for(int i=1;i<=N;i++) X[i] = 0;
    for(int i=n-(1<<k-1);i<n;i++) Y[i-n+(1<<k-1)+1]=i;
    for(int i=1,j=(1<<k-1)+1;i<=n;i++)
        if(sa[i]>=(1<<k-1))Y[j++]=sa[i]-(1<<k-1);
    for(int i=0;i<n;i++) X[rk[i]]++;
    for(int i=1;i<=N;i++) X[i]+=X[i-1];
    for(int i=n;i>=1;i--) sa[X[rk[Y[i]]]--] = Y[i];
    int num = 1; Y[sa[1]] = 1;
    for(int i=2;i<=n;i++) Y[sa[i]] = (chk(i,k-1)?num:++num);
    for(int i=0;i<n;i++) rk[i] = Y[i];
    if(num == n) break;
    }
}
void getheight(){
    for(int i=0;i<n;i++){
    if(i) h[i] = max(0,h[i-1]-1); else h[i] = 0;
    if(rk[i] == 1) continue;
    int comp = sa[rk[i]-1];
    while(str[comp+h[i]] == str[i+h[i]])h[i]++;
    }
    for(int i=0;i<n;i++) height[rk[i]] = h[i];
    for(int i=1;i<=n;i++) RMQ[i][0] = height[i],pos[i][0] = i;
    for(int k=1;(1<<k)<=n;k++){
    for(int i=1;i<=n;i++){
        if(i+(1<<k-1)>n) RMQ[i][k]=RMQ[i][k-1],pos[i][k]=pos[i][k-1];
        else {
        if(RMQ[i][k-1]<RMQ[i+(1<<k-1)][k-1]) pos[i][k] = pos[i][k-1];
        else pos[i][k] = pos[i+(1<<k-1)][k-1];
        RMQ[i][k] = min(RMQ[i][k-1],RMQ[i+(1<<k-1)][k-1]);
        }
    }
    }
}
int getLCP(int L,int R){
    if(L > R) swap(L,R);
    if(L == R) return n-sa[L];
    L++;
    int k = 0; while((1<<k+1)<=R-L+1)k++;
    if(RMQ[L][k]<RMQ[R-(1<<k)+1][k]) return pos[L][k];
    else return pos[R-(1<<k)+1][k];
}

long long ans = 0;

void divide(int l,int r){
    if(l == r) return;
    int ps = getLCP(l,r);
    ans -= 2ll*(ps-l)*(r-ps+1)*height[ps];
    divide(l,ps-1); divide(ps,r);
}

void work(){
    n = strlen(str);
    getsa();
    getheight();
    for(int i=1;i<=n;i++) ans += 1ll*i*i-i;
    for(int i=1;i<=n;i++) ans += 1ll*i*(n-i);
    divide(1,n);
    printf("%lld
",ans);
}

int main(){
    scanf("%s",str);
    work();
    return 0;
}