bzoj3238 [Ahoi2013]差异 后缀数组+单调栈

【bzoj3238】[Ahoi2013]差异

Description

Input

一行，一个字符串S

Output

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

Sample Input

cacao

Sample Output

54

题解：

任意两个字符串的lcp是什么，就是如

a,b  那么若a==b 那么为len(a)

　　否则设sa[a]<sa[b] 那么为min(height[sa[a]+1-------sa[b]])

#include<cstring>
#include<iostream>
#include<cmath>
#include<algorithm>
#include<cstdio>

#define N 500007
#define ll long long
using namespace std;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if (ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
    return x*f;
}

int n;
int stk[N],f[N],g[N];
struct SA
{
    char s[N];
    int a[N],b[N],cnta[N],cntb[N],tsa[N],height[N],sa[N],rk[N*2];
    void Get_SA()
    {
        for (int i=0;i<=256;i++)cnta[i]=0;
        for (int i=1;i<=n;i++)cnta[(int)s[i]]++;
        for (int i=1;i<=256;i++)cnta[i]+=cnta[i-1];
        for (int i=n;i>=1;i--)sa[cnta[(int)s[i]]--]=i;
        rk[sa[1]]=1;
        for (int i=2;i<=n;i++)rk[sa[i]]=rk[sa[i-1]]+(s[sa[i]]!=s[sa[i-1]]);
        for (int i=1;rk[sa[n]]!=n;i<<=1)
        {
            for (int j=1;j<=n;j++)a[j]=rk[j],b[j]=rk[j+i];
            for (int j=0;j<=n;j++)cnta[j]=cntb[j]=0;
            for (int j=1;j<=n;j++)cnta[a[j]]++,cntb[b[j]]++;
            for (int j=1;j<=n;j++)cnta[j]+=cnta[j-1],cntb[j]+=cntb[j-1];
            for (int j=n;j>=1;j--)tsa[cntb[b[j]]--]=j;
            for (int j=n;j>=1;j--)sa[cnta[a[tsa[j]]]--]=tsa[j];
            rk[sa[1]]=1;
            for (int j=2;j<=n;j++)
                rk[sa[j]]=rk[sa[j-1]]+(a[sa[j]]!=a[sa[j-1]]||b[sa[j]]!=b[sa[j-1]]);
        }
    }
    void Get_Height()
    {
        int len=0;
        for (int i=1;i<=n;i++)
        {
            if (len)len--;
            while(s[i+len]==s[sa[rk[i]-1]+len])len++;
            height[rk[i]]=len;
        }
    }
}S;
int main()
{
    scanf("%s",S.s+1);
    n=strlen(S.s+1);
    ll ans=0;
    for (int i=1;i<=n;i++)
    {
        ans+=(ll)(i-1)*i;
        ans+=(ll)i*(i-1)/2;
    }
    S.Get_SA();
    S.Get_Height();
    int tot=0;
    for (int i=2;i<=n;i++)
    {
        while(tot>0&&S.height[i]<S.height[stk[tot]])
            f[stk[tot--]]=i-1;
        stk[++tot]=i;
    }
    while(tot)f[stk[tot--]]=n;
    tot=0;
    for (int i=n;i>=2;i--)
    {
        while(tot>0&&S.height[i]<=S.height[stk[tot]])g[stk[tot--]]=i+1;
        stk[++tot]=i;
    }
    while(tot)g[stk[tot--]]=2;
    for (int i=2;i<=n;i++)
        ans-=(ll)S.height[i]*(ll)(f[i]-i+1)*(ll)(i-g[i]+1)*2;
    printf("%lld
",ans);
}