# P4248 [AHOI2013]差异 （后缀数组+单调栈）

P4248 [AHOI2013]差异 （后缀数组+单调栈）

题目链接：传送门

正文：

首先对于这个公式的前两项我们可以快速求出，为$(1+n)\ast n\ast (n-1)/2$。所以我们只需考虑最后一项的和，又因为每个$T_i$对应排序后的后缀的$rank[i]$位置，所以这步可以转化为在排序后的后缀数组上统计答案。

首先最后一项$lcp(T_i,T_j)$ 的值，设$x=rank[i],y=rank[j]$，不访假设$x<y$，那么$lcp(T_i,T_j)$等于 m i n { h e i g h t [ x + 1 ] , h e i g h t [ x + 2 ] . . . . , h e i g h t [ y ] } min\{height[x+1],height[x+2]....,height[y]\} min{height[x+1],height[x+2]....,height[y]}，即区间$[x+1,y]$的$height[]$数组中的最小的值$height[k]$ 的值是贡献。如果最小高度有多个，那么对规定最左边的最小高度起贡献。

因为最后一项的是所有元组，所以我们可以用单调栈维护每个$height[]$贡献的区间即可，

代码：

#include<bits/stdc++.h>
#define mset(a,b)   memset(a,b,sizeof(a))
typedef long long ll;
const int N=5e5+10;//下标从1开始
int rak[N],sa[N],tp[N],c[N],height[N];
//最后的rak[i]=p 与sa[p]=i ,11对应
void SA(char *s,int n)//这个基数排序的版本中桶的大小为第一关键词rak的最大值
{
    int m=128;//桶的大小,会慢慢变大,最大为n,但初始时是字符的最大值
    //首先基数排序初始化rak数组和sa数组
    //第一轮基数排序,如果s的最大值很大,可改用快速排序
    for(int i=0; i<=m; ++i) c[i]=0;
    for(int i=1; i<=n; ++i) c[rak[i]=s[i]]++;
    for(int i=1; i<=m; ++i) c[i]+=c[i-1];
    for(int i=n; i>=1; --i) sa[c[s[i]]--]=i; //所有后缀长度为1的字符串已求出

    for(int k=1,p ; k < n; k<<=1 ) //p是一个计数器，现在还没用。
    {
        /*tp[i]:第二关键词排名为i的位置为tp[i]*/
        p=0;
        for(int i=n-k+1; i<=n; ++i) tp[++p]=i;
        for(int i=1; i<=n; ++i) if(sa[i]>k) tp[++p]=sa[i]-k;
        /*基数排序求出2k长度的sa数组*/
        for(int i=0; i<=m; ++i) c[i]=0;
        for(int i=1; i<=n; ++i) c[rak[i]]++;
        for(int i=1; i<=m; ++i) c[i]+=c[i-1];
        for(int i=n; i>=1; --i) sa[ c[rak[tp[i]]]-- ]=tp[i];
        /*求该轮的rank数组*/
        //tp才是上一轮的rak,现在要利用上轮的rak和这轮的sa求这轮的rank
        std::swap(rak,tp);
        rak[sa[1]]=p=1;
        for(int i=2,a,b; i<=n; ++i)
        {
            a=sa[i],b=sa[i-1];
            if(tp[a]==tp[b]&&((a+k<=n&& b+k<=n&&tp[a+k]==tp[b+k])||(a+k > n&& b+k > n))) rak[a]=p;
            else rak[a]=++p;
        }
        if(p>=n) break;//可以提前退出
        m=p;
    }//计算高度
    int k=0;
    for(int i=1; i<=n; ++i)
    {
        if(k) k--;//i  ,
        int j=sa[rak[i]-1];
        if(j==0) continue;
        while(s[j+k]==s[i+k]) ++k;
        height[rak[i]]=k;
    }
}
char s[N];
int S[N],top;
int left[N],right[N];
int main()
{
    scanf("%s",s+1);
    int ls=strlen(s+1);
    SA(s,ls);
    top=0;
    for(int i=2;i<=ls;++i)
    {
        while(top > 0&& height[S[top]]> height[i]) top--;
        S[++top]=i;
        if(top==1) left[i]=2;
        else left[i]=S[top-1]+1;
    }
    top=0;
    for(int i=ls;i>=2;--i)
    {
        while(top>0 && height[S[top]] >=height[i]) top--;
        S[++top]=i;
        if(top==1) right[i]=ls;
        else right[i]=S[top-1]-1;
    }
    long long ans=(1ll+ls)*(ls-1)*(ls)/2ll;
    for(int i=2;i<=ls;++i)
    {
        ans-=2ll*height[i]*(right[i]-i+1)*(i-left[i]+1);
    }
    printf("%lld\n",ans);
}