BZOJ 4199: [Noi2015]品酒大会 [后缀数组带权并查集]

4199: [Noi2015]品酒大会

UOJ：http://uoj.ac/problem/131

一年一度的“幻影阁夏日品酒大会”隆重开幕了。大会包含品尝和趣味挑战两个环节，分别向优胜者颁发“首席品酒家”和“首席猎手”两个奖项，吸引了众多品酒师参加。

在大会的晚餐上，调酒师 Rainbow 调制了

在品尝环节上，品酒师 Freda 轻松地评定了每一杯酒的美味度，凭借其专业的水准和经验成功夺取了“首席品酒家”的称号，其中第

输入格式

输入文件的第

第

输出格式

输出文件包括

ans1[L]+=size[x]*size[y];
ans2[L]=max(ans2[L],max(mn[x]*mn[y],mx[x]*mx[y]));

注意i+1相似也是i相似，所以ans要处理后缀和

一点经验：

并查集直接用的是排序后编号

把相同LCP的放在一起可以用邻接链表

注意：

预处理时mx[i]=mn[i]=a[sa[i]]！！！

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
const int N=3e5+5;
typedef long long ll;
const ll INF=1e18+5;
inline int read(){
    char c=getchar();int x=0,f=1;
    while(c<'0'||c>'9'){if(c=='-')f=-1; c=getchar();}
    while(c>='0'&&c<='9'){x=x*10+c-'0'; c=getchar();}
    return x*f;
}
int n,a[N];
char s[N];
struct edge{
    int v,ne;
}e[N];
int cnt,h[N];
inline void ins(int u,int v){
    cnt++;
    e[cnt].v=v;e[cnt].ne=h[u];h[u]=cnt;
}

int sa[N],rnk[N],t1[N],t2[N],height[N],c[N];
bool cmp(int *r,int a,int b,int j){
    return a+j<=n&&b+j<=n&&r[a]==r[b]&&r[a+j]==r[b+j];
}
void getSA(int m){
    int *r=t1,*k=t2;
    for(int i=1;i<=n;i++) c[r[i]=s[i]]++;
    for(int i=1;i<=m;i++) c[i]+=c[i-1];
    for(int i=n;i>=1;i--) sa[c[r[i]]--]=i;
    for(int j=1;j<=n;j<<=1){
        int p=0;
        for(int i=n-j+1;i<=n;i++) k[++p]=i;
        for(int i=1;i<=n;i++) if(sa[i]>j) k[++p]=sa[i]-j;
        
        for(int i=0;i<=m;i++) c[i]=0;
        for(int i=1;i<=n;i++) c[r[k[i]]]++;
        for(int i=1;i<=m;i++) c[i]+=c[i-1];
        for(int i=n;i>=1;i--) sa[c[r[k[i]]]--]=k[i];
        
        swap(r,k);p=0;r[sa[1]]=++p;
        for(int i=2;i<=n;i++) r[sa[i]]=cmp(k,sa[i],sa[i-1],j)?p:++p;
        if(p>=n) break;m=p;
    }
}
void getHeight(){
    for(int i=1;i<=n;i++) rnk[sa[i]]=i;
    int k=0;
    for(int i=1;i<=n;i++){
        if(k) k--;
        if(rnk[i]==1) continue;
        int j=sa[rnk[i]-1];
        while(i+k<=n&&j+k<=n&&s[i+k]==s[j+k]) k++;
        height[rnk[i]]=k;
    }
}

int fa[N],size[N];
ll ans1[N],ans2[N],mn[N],mx[N];
inline int find(int x){return x==fa[x]?x:fa[x]=find(fa[x]);}
void unn(int x,int y,int L){//printf("unn %d %d %d\n",x,y,L);
    x=find(x);y=find(y);
    if(x==y) return;
    //if(x<y) swap(x,y);
    ans1[L]+=(ll)size[x]*size[y];
    ans2[L]=max(ans2[L],max(mn[x]*mn[y],mx[x]*mx[y]));
    fa[x]=y;
    mx[y]=max(mx[y],mx[x]);
    mn[y]=min(mn[y],mn[x]);
    size[y]+=size[x];    
}
void solve(){
    for(int i=1;i<=n;i++){
        fa[i]=i,size[i]=1,mx[i]=mn[i]=a[sa[i]],ans2[i]=-INF;
        if(i!=1) ins(height[i],i);
    }
    for(int l=n-1;l>=0;l--){
        for(int i=h[l];i;i=e[i].ne) unn(e[i].v-1,e[i].v,l);
        if(l!=n-1){
            ans1[l]+=ans1[l+1];
            ans2[l]=max(ans2[l],ans2[l+1]);
        }
    }
    for(int i=0;i<=n-1;i++) printf("%lld %lld\n",ans1[i],ans2[i]==-INF?0:ans2[i]);
}
int main(){
    //freopen("in.txt","r",stdin);
    n=read();
    scanf("%s",s+1);
    for(int i=1;i<=n;i++) a[i]=read();
    getSA(256);
    getHeight();
    solve();        
}