[UOJ 131]【NOI2015】品酒大会：后缀数组+并查集

点击这里查看原题

首先用后缀数组求出每个串和上个串的公共前缀长度，然后按公共前缀长度由长到短进行合并，用并查集维护。
注意：关闭了流同步的cin读入字符串依然会TLE

/*
User:Small
Language:C++
Problem No.:131
*/
#include<bits/stdc++.h>
#define ll long long
#define inf ((ll)1<<60)
using namespace std;
const int M=3e5+5;
int n,v[M],rk[2][M],sa[2][M],p,q,a[M],b[M],h[M],lis[M],pre[M],siz[M],mn[M],mx[M];
ll ans1[M],ans2[M];
char s[M];
void calsa(int k,int sa[],int rk[],int SA[],int RK[]){
    for(int i=1;i<=n;i++) v[rk[sa[i]]]=i;
    for(int i=n;i;i--)
        if(sa[i]>k) SA[v[rk[sa[i]-k]]--]=sa[i]-k;
    for(int i=n-k+1;i<=n;i++) SA[v[rk[i]]--]=i;
    for(int i=1;i<=n;i++) RK[SA[i]]=RK[SA[i-1]]+(rk[SA[i]]!=rk[SA[i-1]]||rk[SA[i]+k]!=rk[SA[i-1]+k]);
}
void work(){
    p=0,q=1;
    for(int i=1;i<=n;i++) a[i]=s[i-1]-'a'+1;
    for(int i=1;i<=n;i++) v[a[i]]++;
    for(int i=1;i<=30;i++) v[i]+=v[i-1];
    for(int i=1;i<=n;i++) sa[p][v[a[i]]--]=i;
    for(int i=1;i<=n;i++) rk[p][sa[p][i]]=rk[p][sa[p][i-1]]+(a[sa[p][i]]!=a[sa[p][i-1]]);
    for(int k=1;k<n;k<<=1){
        calsa(k,sa[p],rk[p],sa[q],rk[q]);
        swap(p,q);
    }
}
void geth(){
    int k=0;
    for(int i=1;i<=n;i++){
        if(rk[p][i]==1) continue;
        int j=sa[p][rk[p][i]-1];
        while(a[i+k]==a[j+k]) k++;
        h[rk[p][i]]=k;
        if(k) k--;
    }
}
bool cmp(int x,int y){
    return h[x]!=h[y]?h[x]>h[y]:x<y;
}
int find(int x){
    return x==pre[x]?x:pre[x]=find(pre[x]);
}
void merge(int x,int y){
    pre[x]=y;
    siz[y]+=siz[x];
    mx[y]=max(mx[y],mx[x]);
    mn[y]=min(mn[y],mn[x]);
}
int main(){
    freopen("data.in","r",stdin);//
    scanf("%d%s",&n,s);
    for(int i=1;i<=n;i++) scanf("%d",&b[i]);
    work();
    geth();
    for(int i=1;i<=n;i++){
        siz[i]=1;
        pre[i]=i;
        mn[i]=mx[i]=b[sa[p][i]];
    }
    for(int i=1;i<n;i++) lis[i]=i+1;
    sort(lis+1,lis+n,cmp);
    for(int i=0;i<n;i++) ans2[i]=-inf;
    for(int i=1;i<n;i++){
        int x=find(lis[i]-1),y=find(lis[i]);
        ans1[h[lis[i]]]+=(ll)siz[x]*siz[y];
        ans2[h[lis[i]]]=max(ans2[h[lis[i]]],(ll)mx[x]*mx[y]);
        ans2[h[lis[i]]]=max(ans2[h[lis[i]]],(ll)mn[x]*mx[y]);
        ans2[h[lis[i]]]=max(ans2[h[lis[i]]],(ll)mx[x]*mn[y]);
        ans2[h[lis[i]]]=max(ans2[h[lis[i]]],(ll)mn[x]*mn[y]);
        merge(x,y);
    }
    for(int i=n-2;i>=0;i--){
        ans1[i]+=ans1[i+1]; 
        ans2[i]=max(ans2[i],ans2[i+1]);
    }
    for(int i=0;i<n;i++){
        printf("%lld %lld\n",ans1[i],(ans1[i]?ans2[i]:0));
    }
    return 0;
}