bzoj 3238: [Ahoi2013]差异 -- 后缀数组

3238: [Ahoi2013]差异

Time Limit: 20 Sec  Memory Limit: 512 MB

Description

Input

一行，一个字符串S

Output

一行，一个整数，表示所求值

Sample Input

cacao

Sample Output

54

HINT

2<=N<=500000,S由小写英文字母组成

Source

后缀数组+单调栈水过。。。

#include<map>
#include<cmath>
#include<queue>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define inf 1000000007
#define ll long long
#define N 500010
int v[N],rk1[N],rk2[N],sa1[N],sa2[N],n,a[N],ht[N];
int l[N],r[N],q[N],tot;
int *sa=sa1,*SA=sa2,*rk=rk1,*RK=rk2;
char s[N];
ll ans;
int main()
{
    scanf("%s",s+);
    n=strlen(s+);
    for(int i=;i<=n;i++){a[i]=s[i]-'a'+;v[a[i]]++;}
    for(int i=;i<=;i++) v[i]+=v[i-];
    for(int i=n;i;i--) sa[v[a[i]]--]=i;
    for(int i=;i<=n;i++) rk[sa[i]]=rk[sa[i-]]+(a[sa[i]]!=a[sa[i-]]);
    for(int k=;k<=n;k<<=)
    {
        for(int i=;i<=n;i++) v[rk[sa[i]]]=i;
        for(int i=n;i;i--) if(sa[i]>k) SA[v[rk[sa[i]-k]]--]=sa[i]-k;
        for(int i=n-k+;i<=n;i++) SA[v[rk[i]]--]=i;
        for(int i=;i<=n;i++) RK[SA[i]]=RK[SA[i-]]+(rk[SA[i]]!=rk[SA[i-]]||rk[SA[i]+k]!=rk[SA[i-]+k]);
        swap(sa,SA);swap(rk,RK);
    }
    int k=,j;
    for(int i=;i<n;i++)
    {
        j=sa[rk[i]-];
        while(s[i+k]==s[j+k]) k++;
        ht[rk[i]]=k;
        if(k) k--;
    }
    ht[]=-inf;
    for(int i=;i<=n;i++)
    {
        while(ht[i]<=ht[q[tot]]) tot--;
        l[i]=q[tot]+; q[++tot]=i;
    }
    ht[n+]=-inf;tot=;q[]=n+;
    for(int i=n;i;i--)
    {
        while(ht[i]<ht[q[tot]]) tot--;
        r[i]=q[tot]-; q[++tot]=i;
    }
    ans=(ll)(n+)*n*(n-)/;
    for(int i=;i<=n;i++) ans-=2LL*(i-l[i]+)*(r[i]-i+)*ht[i];
    printf("%lld\n",ans);
    return ;
}