BZOJ 4199 [Noi2015]品酒大会

后缀数组好题，把height从大到小排序，利用并查集维护每个height值大于0的后缀。由于权值可能为负，所以维护最大值的同时还要维护最小值，每次合并左右相邻rank

ps:有抄代码的嫌疑…..

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const int maxn=305000;
int n,m;
char s[maxn];
int t1[maxn],t2[maxn],c[maxn],sa[maxn],rank[maxn],val[maxn],f[maxn],L[maxn],R[maxn],mn[maxn],mx[maxn];
ll ans[maxn],cnt[maxn];
struct node
{
    ll val,id;      
};
node height[maxn];
void build_sa(int m,int n)
{
    int *x=t1,*y=t2;
    for(int i=0;i<m;i++) c[i]=0;
    for(int i=0;i<n;i++) c[x[i]=s[i]]++;
    for(int i=0;i<m;i++) c[i]+=c[i-1];
    for(int i=n-1;i>=0;i--) sa[--c[x[i]]]=i;
    for(int k=1;k<=n;k<<=1)
    {
        int p=0;
        for(int i=n-k;i<n;i++) y[p++]=i;
        for(int i=0;i<n;i++) if(sa[i]>=k) y[p++]=sa[i]-k;
        for(int i=0;i<m;i++) c[i]=0;
        for(int i=0;i<n;i++) c[x[y[i]]]++;
        for(int i=0;i<m;i++) c[i]+=c[i-1];
        for(int i=n-1;i>=0;i--) sa[--c[x[y[i]]]]=y[i];
        swap(x,y);
        p=1;x[sa[0]]=0;
        for(int i=1;i<n;i++) x[sa[i]]= y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k] ? p-1 :p++;
        if(p>=n) break;
        m=p;
    }
}
/*void get_height()
{
    int j,k=0;
    for(int i=0;i<n;i++) rank[sa[i]]=i;
    for(int i=0;i<n;i++)
    {
        if(k) k--;
        if(!rank[i]) continue;
        j=sa[rank[i]-1];
        while(s[j+k]==s[i+k]) k++;
        height[rank[i]]=(node){k,rank[i]};
    }
}*/
void get_height()
{
    for (int i=1;i<=n;i++) rank[sa[i]]=i;
    int k=0,x;
    for (int i=0;i<n;i++)
    {
        k=max(k-1,0);
        x=sa[rank[i]-1];
        while (s[i+k]==s[x+k]) k++;
        height[rank[i]]=(node){k,rank[i]};
    }
} 
int find(int x)
{
    if(f[x]==x) return x;
    else return f[x]=find(f[x]);
}
bool cmp(const node &i,const node &j)
{
    return i.val>j.val;
}
int main()
{
    scanf("%d",&n);
    scanf("%s",s);
    build_sa(256,n+1);
    get_height();
    memset(ans,-127,sizeof(ans));
    for(int i=0;i<n;i++) scanf("%d",&val[i]);
    sort(height+1,height+n+1,cmp);
    for(int i=1;i<=n;i++) f[i]=i,L[i]=i,R[i]=i,mn[i]=mx[i]=val[sa[i]];
    for(int i=1;i<=n;i++)
    if(!height[i].val) break;
    else
    {
        int fx=find(height[i].id),fy=find(height[i].id-1);
        cnt[height[i].val]+=(ll)(R[fx]-L[fx]+1)*(R[fy]-L[fy]+1);
        ans[height[i].val]=max((ll)mn[fx]*mn[fy],max(ans[height[i].val],(ll)mx[fx]*mx[fy])); 
        f[fx]=fy;
        L[fy]=min(L[fx],L[fy]);R[fy]=max(R[fx],R[fy]);
        mn[fy]=min(mn[fx],mn[fy]);
        mx[fy]=max(mx[fx],mx[fy]);
    }
    for(int i=n-1;i>=1;i--)
        cnt[i]+=cnt[i+1],ans[i]=max(ans[i],ans[i+1]);
    sort(val,val+n);
    printf("%lld %lld\n",(ll)n*(n-1)>>1,max((ll)val[1]*val[0],(ll)val[n-2]*val[n-1]));
    for(int i=1;i<n;i++)
    {       
        printf("%lld %lld\n",cnt[i],! cnt[i] ? 0 :ans[i]);
    }
    return 0;
}