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

点击这里查看原题

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

/*User:SmallLanguage: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;}