【AHOI2013】bzoj3238 差异

来源：互联网发布：海外淘宝网编辑：程序博客网时间：2024/06/05 02:55

实际上只需要求出后缀两两之间的lcp之和。也就是求height数组所有区间的区间最小值之和。这样扫描一遍，用单调栈维护区间最小height值递增的序列的左端点即可，建出sa以后是O(n)的。

#include<cstdio>#include<cstring>#include<algorithm>using namespace std;#define LL long longconst int maxn=500010;char s[maxn];int sa[maxn],rank[maxn],height[maxn],f[maxn],sta[maxn],val[maxn],cnt[maxn],n;int main(){    int p,m=26,top;    LL ans=0,now=0;    scanf("%s",s+1);    n=strlen(s+1);    for (int i=1;i<=n;i++) cnt[rank[i]=s[i]-'a'+1]++;    for (int i=2;i<=m;i++) cnt[i]+=cnt[i-1];    for (int i=n;i;i--) sa[cnt[rank[i]]--]=i;    for (int k=1;;k<<=1)    {        p=0;        for (int i=n-k+1;i<=n;i++) f[++p]=i;        for (int i=1;i<=n;i++)            if (sa[i]>k) f[++p]=sa[i]-k;        for (int i=1;i<=m;i++) cnt[i]=0;        for (int i=1;i<=n;i++) cnt[rank[f[i]]]++;        for (int i=2;i<=m;i++) cnt[i]+=cnt[i-1];        for (int i=n;i;i--) sa[cnt[rank[f[i]]]--]=f[i];        for (int i=1;i<=n;i++) f[i]=rank[i];        rank[sa[1]]=1;        for (int i=2;i<=n;i++)            if (f[sa[i]]!=f[sa[i-1]]||f[sa[i]+k]!=f[sa[i-1]+k])                rank[sa[i]]=rank[sa[i-1]]+1;            else                rank[sa[i]]=rank[sa[i-1]];        m=rank[sa[n]];        if (m>=n) break;    }    for (int i=1;i<=n;i++)    {        height[rank[i]]=height[rank[i-1]];        if (height[rank[i]]) height[rank[i]]--;        while (s[i+height[rank[i]]]==s[sa[rank[i]-1]+height[rank[i]]])            height[rank[i]]++;    }    sta[top=1]=1;    for (int i=2;i<=n;i++)    {        p=i;        while (top&&height[i]<=val[top])        {            now-=(LL)(p-sta[top])*(val[top]-height[i]);            p=sta[top];            top--;        }        sta[++top]=p;        val[top]=height[i];        now+=height[i];        ans-=now;    }    //printf("%lld\n",-ans);    ans*=2;    for (int i=1;i<=n;i++) ans+=(LL)3*i*(i-1)/2;    printf("%lld\n",ans);}

0 0