bzoj4812: [Ynoi2017]由乃打扑克

由于查询的是树链的并的信息，同时信息不能高效合并，只能考虑用bitset维护，小范围暴力预处理以便从bitset算出答案

对树分块，保证每块是连通的且直径较小，对分出的块缩点建新树，在新树上建树上ST表，用bitset保存信息，于是每条链只需用4个bitset合并，再暴力加上零散部分

考虑到bitset的复杂度非常高，分块部分有很大的调整空间，不会成为瓶颈

#include<cstdio>
#include<cstring>
const int M=,N=,N4=N*,B=;
char buf[M*],*ptr=buf-;
int _(){
    int x=,c=*++ptr;
    while(c<)c=*++ptr;
    while(c>)x=x*+c-,c=*++ptr;
    return x;
}
typedef unsigned int bits[N];
void _or(bits a,bits b){for(int i=;i<N;i+=)b[i]|=a[i],b[i+]|=a[i+];}
void _set(bits a,int x){a[x>>]|=<<x;}
int n,m,la=,es[*M],enx[*M],e0[M],ep=,v[M];
int fa[M],sz[M],top[M],dep[M],son[M];
bits ans;
void f1(int w,int pa){
    dep[w]=dep[fa[w]=pa]+;
    sz[w]=;
    for(int i=e0[w];i;i=enx[i]){
        int u=es[i];
        if(u!=pa){
            f1(u,w);
            sz[w]+=sz[u];
            if(sz[u]>sz[son[w]])son[w]=u;
        }
    }
}
void f2(int w,int tp){
    top[w]=tp;
    if(son[w])f2(son[w],tp);
    for(int i=e0[w];i;i=enx[i]){
        int u=es[i];
        if(u!=fa[w]&&u!=son[w])f2(u,u);
    }
}
int lca(int x,int y){
    int a=top[x],b=top[y];
    while(a!=b){
        if(dep[a]>dep[b])x=fa[a],a=top[x];
        else y=fa[b],b=top[y];
    }
    return dep[x]<dep[y]?x:y;
}
void maxs(int&a,int b){if(a<b)a=b;}
int id[M],idp=,md[M],rt[M],fas[M/B][],deps[M],e1[M];
bits st[M/B][];
void f5(int w){
    for(int i=e1[w];i;i=enx[i]){
        int u=es[i];
        deps[u]=deps[w]+;
        f5(u);
    }
}
void f4(int w){
    id[w]=idp;
    for(int i=e0[w];i;i=enx[i]){
        int u=es[i];
        if(u!=fa[w]&&!id[u])f4(u);
    }
}
void f3(int w){
    for(int i=e0[w];i;i=enx[i]){
        int u=es[i];
        if(u!=fa[w]){
            f3(u);
            if(!id[u])maxs(md[w],md[u]+);
        }
    }
    if(w==||md[w]==B){
        rt[++idp]=w;
        f4(w);
    }
}
int log_2[M];
void cal(int x,int y){
    int D=deps[id[x]]-deps[id[y]]-;
    if(D<=){
        _set(ans,v[y]);
        while(x!=y)_set(ans,v[x]),x=fa[x];
        return;
    }
    int a=rt[id[x]];
    while(x!=a)_set(ans,v[x]),x=fa[x];
    x=id[x];
    int d=log_2[D];
    _or(st[x][d],ans);
    D-=<<d;
    if(D){
        for(int i=;i<;++i)if(D>>i&)x=fas[x][i];
        _or(st[x][d],ans);
    }
    x=rt[fas[x][d]];
    while(x!=y)x=fa[x],_set(ans,v[x]);
}
int ls[],rs[],ms[][];
int pw[][];
int xs[N4+];
int _cal(bits x,int k){
    int s=,xp=;
    for(int i=;i<N;++i){
        xs[xp]=x[i]&;
        xs[xp+]=x[i]>>&;
        xs[xp+]=x[i]>>&;
        xs[xp+]=x[i]>>&;
        xp+=;
    }
    for(int i=,d=;i<N4;++i){
        while(xs[i]==)++i,d+=;
        s+=pw[k][ls[xs[i]]+d];
        s+=ms[k][xs[i]];
        d=rs[xs[i]];
    }
    return s;
}
int main(){
    fread(buf,,sizeof(buf),stdin)[buf]=;
    for(int i=;i<=;++i)pw[][i]=;
    for(int t=;t<=;++t)
    for(int i=;i<=;++i)pw[t][i]=pw[t-][i]*i;
    for(int i=;i<;++i){
        int vs[],vp=;
        for(int a=,b=;a<;a=b){
            b=a;
            if(~i>>a&){
                ++b;
                continue;
            }
            while(i>>b&)++b;
            vs[vp++]=b-a;
        }
        if(i&)ls[i]=vs[];
        if(i>>&)rs[i]=vs[vp-];
        for(int j=(i&);j<vp-(i>>&);++j){
            for(int k=;k<=;++k)ms[k][i]+=pw[k][vs[j]];
        }
    }
    n=_();m=_();
    for(int i=;i<=n;++i)v[i]=_();
    for(int i=,a,b;i<n;++i){
        a=_();b=_();
        es[ep]=b;enx[ep]=e0[a];e0[a]=ep++;
        es[ep]=a;enx[ep]=e0[b];e0[b]=ep++;
    }
    f1(,);f2(,);
    f3();
    for(int i=;i<=idp;++i){
        int w=rt[i],u=fa[rt[fas[i][]=id[fa[rt[i]]]]];
        while(w!=u)_set(st[i][],v[w]),w=fa[w];
    }
    for(int i=;i<;++i){
        for(int j=;j<=idp;++j){
            int k=fas[j][i-];
            if(fas[j][i]=fas[k][i-]){
                memcpy(st[j][i],st[j][i-],sizeof(bits));
                _or(st[k][i-],st[j][i]);
            }
        }
    }
    log_2[]=-;
    for(int i=;i<=idp;++i){
        log_2[i]=log_2[i>>]+;
        if(fas[i][]){
            int a=fas[i][];
            es[ep]=i;enx[ep]=e1[a];e1[a]=ep++;
        }
    }
    f5(idp);
    while(m--){
        memset(ans,,sizeof(ans));
        for(int c=_();c;--c){
            int x=_()^la,y=_()^la,z=lca(x,y);
            cal(x,z);
            cal(y,z);
        }
        printf("%u\n",la=_cal(ans,_()));
    }
    return ;
}