【BZOJ3238】差异 后缀数组

反正又要办停课了不如一块一块慢慢刷。。。最近集中处理一下字符串和数据结构（说起来真轻巧）

这种直接把“后缀”两个字糊脸上的题也是不常见。

首先长度和（也就是求和式中的前两项）比较简单，不讨论，重点就是原来字符串中任意两项的公共前缀如何计算。

首先将sa数组和height数组做出来（不要和h搞混，注意弄清含义），然后计算 以height[i]作为答案的数对的组数 K ，在答案中减去2*k*height[i]即可。

问题变为如何算k，注意到如果要以height[i]作为答案，一定有一个的rank<=i，而另一个的 rank>=i，因此我们预处理出两个数组L,R，分别表示在height数组中，在i之前并且height小于height[i]的最后一个元素，在i之后并且height小于等于height[i]的最前一个元素，k=(i-L[i])*(R[i]-i);

问题变为如何算L和R，此时我们可以用一个单调栈来维护，保证栈中的元素递增并且小于（或小于等于）当前元素，正反各扫一遍即可。

有关小于和小于等于的问题，LR一边一种是保证height值相同的情况下不会算重或者算漏。

/**************************************************************
    Problem: 3238
    User: RicardoWang
    Language: C++
    Result: Accepted
    Time:4444 ms
    Memory:31548 kb
****************************************************************/
 
#include<cstdlib>
#include<cstdio>
#include<iostream>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
#include<queue>
using namespace std;
#define maxn 500005
char s[2*maxn];
int len,n,m;
void Init()
{
     scanf("%s",s+1);
     for(len=1;;len++)
     {
        if(s[len]=='\0'||s[len]=='\n')
        {
            len--; break;
        }
     }
     n=len;
     m=500;
     return ;
}
int wa[maxn*2],wb[maxn*2],sa[maxn*2],cc[maxn*2],rank[2*maxn],height[2*maxn];
void makesa()
{
    int *y=wa,*x=wb,*t;
    int cnt;
    for(int i=1;i<=n;i++)x[i]=s[i];
    for(int i=1;i<=n;i++)cc[x[i]]++;
    for(int i=1;i<=m;i++)cc[i]=cc[i]+cc[i-1];
    for(int i=n;i>=1;i--)sa[cc[x[i]]--]=i;
    for(int k=1;k<=n;k=k*2)
    {
        cnt=0;
        for(int i=n-k+1;i<=n;i++)y[++cnt]=i;
        for(int i=1;i<=n;i++)if(sa[i]>k)y[++cnt]=sa[i]-k;
        for(int i=1;i<=m;i++)cc[i]=0;
        for(int i=1;i<=n;i++)cc[x[y[i]]]++;
        for(int i=1;i<=m;i++)cc[i]=cc[i]+cc[i-1];
        for(int i=n;i>=1;i--)sa[cc[x[y[i]]]--]=y[i];
        t=x; x=y; y=t;x[sa[1]]=1;cnt=1;
        for(int i=2;i<=n;i++)
        {
            x[sa[i]]=(y[sa[i]]==y[sa[i-1]] && y[sa[i]+k]==y[sa[i-1]+k])? cnt:++cnt;
        }
        if(cnt==n)break; m=cnt;
    }
    return ;
}
void calcrank()
{
    int k=0;
    for(int i=1;i<=n;i++)rank[sa[i]]=i;
    for(int i=1;i<=n;i++)
    {
        if(k)k--;
        for(int j=sa[rank[i]-1];s[j+k]==s[i+k];k++);
        height[rank[i]]=k;
    }
    return ;
}
int L[maxn],R[maxn],sta[maxn],top;
long long ans;
void work()
{
    top=0;
    for(int i=1;i<=n;i++)
    {
        while(top && height[sta[top]]>=height[i])top--;
        if(!top)L[i]=1; else L[i]=sta[top]+1;
        sta[++top]=i;
    }
    top=0;
    for(int i=n;i>=1;i--)
    {
        while(top && height[sta[top]]>height[i])top--;
        if(!top)R[i]=n; else R[i]=sta[top]-1;
        sta[++top]=i;
    }
    ans=(long long)n*(long long)(n-1)*(long long)(n+1)/2;
    for(int i=1;i<=n;i++)
    {
        ans=ans-(long long )height[i]*(i-L[i]+1)*(R[i]-i+1)*2;
    }
    cout<<ans<<endl;
    return ;
}
int main()
{
    //freopen("in.txt","r",stdin);
    Init();
    makesa();
    calcrank();
    work();
    return 0;
}