BZOJ 3238: [Ahoi2013]差异 后缀数组

卡时过得，感觉自己用错方法了，算了，练练后缀数组也是挺好的。
我们只需要求出height数组之后跑一边单调栈就能知道以每一个点为最小值能扩张的最远的区间，用总的值减去2*左边的个数*右边的个数*height[i]就可以了
[注意] 1.为了防止区间发生重叠，向左面扩展时相同不扩展，向右面扩展时相同也扩展
2.要弄清楚height的具体含义，搞清边界问题

#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<iostream>
#include<iomanip>
#include<algorithm>
#include<ctime>
#include<string>
using namespace std;
int sa[1000000];
int rnk[1000000];
int x[1000000];
int y[1000000];
int temp[1000000];
int sum[1000000];
char s[1000000];
int height[1000000];
int len;
int tot;
void get_rank()
{
    for(int i=1;i<=len;i++) sum[s[i]]++;
    for(int i=1;i<=127;i++) sum[i]+=sum[i-1];
    for(int i=len;i>=1;i--) temp[sum[s[i]]--]=i;
    tot=0;
    for(int i=1;i<=len;i++)
    {
        if(i==1 || s[temp[i]]!=s[temp[i-1]]) tot++;
        rnk[temp[i]]=tot;
    }
}
void ji_sort(int key[],int order[])
{
    for(int i=0;i<=len;i++) sum[i]=0;
    for(int i=1;i<=len;i++) sum[key[i]]++;
    for(int i=1;i<=len;i++) sum[i]+=sum[i-1];
    for(int i=len;i>=1;i--) temp[sum[key[order[i]]]--]=order[i];
    for(int i=1;i<=len;i++) order[i]=temp[i];
}

void get_hou()
{
    get_rank();
    for(int j=1;j<=len;j<<=1)
    {
        for(int i=1;i<=len;i++)
        {
            x[i]=rnk[i];
            y[i]=i+j>len? 0 :rnk[i+j];
            sa[i]=i;
        }
        ji_sort(y,sa);
        ji_sort(x,sa);
        tot=0;
        for(int i=1;i<=len;i++)
        {
            if(i==1 || x[sa[i]]!=x[sa[i-1]] || y[sa[i]]!=y[sa[i-1]]) tot++;
            rnk[sa[i]]=tot;
        }
    }
}
int get_height()
{
    for(int i=1;i<=len;i++)
    {
        if(rnk[i]==1) continue;
        int j=max(height[rnk[i-1]]-1,0),k=sa[rnk[i]-1];
        while(s[k+j]==s[i+j]) j++;
        height[rnk[i]]=j;
    }
}
int my_right[1000000];
int my_left[1000000];
int my_stack[1000000];
int main()
{
    scanf("%s",s+1);
    len=strlen(s+1);
    get_hou();
    get_height();
    int top=0;
    for(int i=1;i<=len;i++)
    {
        while(height[my_stack[top]]>height[i] && top>0) top--;
        if(top==0) my_left[i]=1;
        else my_left[i]=my_stack[top]+1;
        my_stack[++top]=i;
    }
    top=0;
    for(int i=len;i>=1;i--)
    {
        while(height[my_stack[top]]>=height[i] && top>0) top--;
        if(top==0) my_right[i]=len;
        else my_right[i]=my_stack[top]-1;
        my_stack[++top]=i;
    }
    long long o=len;
    long long ans=(o+1)*o*(o-1)/2;
    long long t;
    for(int i=1;i<=len;i++)
    {
        t=2*(i-my_left[i]+1);
        t*=my_right[i]-i+1;
        t*=height[i];
        ans-=t;
    }
    cout<<ans;
    return 0;
}