NOI 2015 品酒大会 (后缀数组+并查集)

题目大意：略

40分暴力还是很好写的，差分再跑个后缀和 和 后缀最大值就行了

一种正解是后缀数组+并查集

但据说还有后缀数组+单调栈的高端操作蒟蒻的我当然不会

后缀数组求出height，然后从大到小排个序，那么每次把最大height的两个后缀放到一个并查集里

这样，每次根据height合并两个并查集的时候，一个并查集任何一个后缀和另一个并查集任何一个后缀的LCP长度都是height[i]

并查集维护集合内a[i]最大值，最小值(负负得正)，然后每次合并都打差分，统计答案的时候跑一遍后缀和和后缀最大值即可

#include <cmath>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define ll long long 
#define N 301000
#define rint register int
#define inf 1000000000000000001ll
using namespace std;
//re
int len;
char str[N];
int tr[N],rk[N],hs[N],sa[N],fa[N];
ll a[N],ma[N],mi[N],ans[N],sum[N],p[N],sz[N];
struct Height{int id,w;}h[N];
int cmp(Height s1,Height s2){return s1.w>s2.w;}
bool check(int k,int x,int y){
    if(x+k>len||y+k>len) return 0;
    else return (rk[x]==rk[y]&&rk[x+k]==rk[y+k])?1:0;
}
void get_sa()
{
    int cnt=0;rint i;
    for(i=1;i<=len;++i) hs[str[i]]++;
    for(i=1;i<=127;++i) if(hs[i]) tr[i]=++cnt;
    for(i=1;i<=127;++i) hs[i]+=hs[i-1];
    for(i=1;i<=len;++i) rk[i]=tr[str[i]],sa[hs[str[i]]--]=i;
    for(int k=1;cnt<len;k<<=1)
    {
        for(i=1;i<=cnt;++i) hs[i]=0;
        for(i=1;i<=len;++i) hs[rk[i]]++;
        for(i=1;i<=cnt;++i) hs[i]+=hs[i-1];
        for(i=len;i>=1;i--) if(sa[i]>k) tr[sa[i]-k]=hs[rk[sa[i]-k]]--;
        for(i=1;i<=k;++i) tr[len-i+1]=hs[rk[len-i+1]]--;
        for(i=1;i<=len;++i) sa[tr[i]]=i;
        for(i=1,cnt=0;i<=len;++i) tr[sa[i]]=check(k,sa[i],sa[i-1])?cnt:++cnt;
        for(i=1;i<=len;++i) rk[i]=tr[i];
    }
    for(i=1;i<=len;++i){
        h[i].id=i;
        if(rk[i]==1) continue;
        for(int j=max(1,h[rk[i-1]].w-1);;j++)
            if(str[i+j-1]==str[sa[rk[i]-1]+j-1]) h[rk[i]].w=j;
            else break;
    }
}
int find_fa(int x){
    int y=x;    
    while(y!=fa[y]) y=fa[y];
    while(fa[x]!=y){
        int pre=fa[x];
        fa[x]=y;
        x=pre;
    }return y;
}
void solve()
{
    rint i;int x,y,fx,fy;
    for(i=1;i<=len;++i) sz[i]=1,fa[i]=i,mi[i]=ma[i]=a[sa[i]];
    for(i=1;i<=len;)
    {
        int j=h[i].w;
        for(;h[i].w>=j&&i<=len;i++){
            x=h[i].id-1;fx=find_fa(x);
            y=h[i].id;fy=find_fa(y);
            p[j]+=sz[fx]*sz[fy];
            ans[j]=max(ans[j],max(mi[fx]*mi[fy],ma[fx]*ma[fy]));
            fa[fy]=fx,sz[fx]+=sz[fy];
            ma[fx]=max(ma[fx],ma[fy]),mi[fx]=min(mi[fx],mi[fy]);
        }
    }
}

int main()
{
    scanf("%d",&len);
    scanf("%s",str+1);
    for(int i=1;i<=len;++i)
        scanf("%lld",&a[i]);
    memset(ans,-0x3f,sizeof(ans));
    get_sa();
    sort(h+1,h+len+1,cmp);
    solve();
    for(int i=len;i>=0;i--)
        sum[i]=sum[i+1]+p[i],ans[i]=max(ans[i],ans[i+1]);
    for(int i=0;i<len;i++)
        printf("%lld %lld
",sum[i],(ans[i]<=-inf)?0:ans[i]);
    return 0;
}