HDU6271：Master of Connected Component（树分块+并查集）

传送门
题面链接

题解：
$O(n^2 \log n )$就直接暴力并查集。

不过注意到我们要算很多重复的部分。容易想到dfs到一个点，我们把距离他$O(\sqrt{n})$的点的答案都做出来。 合理分配后这样的点最多$\sqrt{n}$个。我们对这$\sqrt{n}$个点都算一次即可。 时间复杂度$O(n \sqrt{n} \log n)$。

#include <bits/stdc++.h>
#include <tr1/unordered_map>
using namespace std;
typedef pair <int,int> pii;

const int RLEN=1<<18|1;
inline char nc() {
    static char ibuf[RLEN],*ib,*ob;
    (ib==ob) && (ob=(ib=ibuf)+fread(ibuf,1,RLEN,stdin));
    return (ib==ob) ? -1 : *ib++;
}
inline int rd() {
    char ch=nc(); int i=0,f=1;
    while(!isdigit(ch)) {if(ch=='-')f=-1; ch=nc();}
    while(isdigit(ch)) {i=(i<<1)+(i<<3)+ch-'0'; ch=nc();}
    return i*f;
}
inline void W(int x) {
    static int buf[50];
    if(!x) {putchar('0'); return;}
    if(x<0) {putchar('-'); x=-x;}
    while(x) {buf[++buf[0]]=x%10; x/=10;}
    while(buf[0]) {putchar(buf[buf[0]--]+'0');}
}

const int N=2e4+50, B=350;
int n,m,rt,fa[N],d[N],G[N],ans[N]; 
int anc[N],rk[N],tot; pii g[N];
vector <int> edge[N];
tr1::unordered_map <int,bool> qry[N];
struct data {
    int x,y,type;
    data() {}
    data(int x,int y,int type) : x(x), y(y), type(type){}
} opt[N];
inline void init() {
    for(int i=1;i<=2*n;i++) {G[i]=0; edge[i].clear(); fa[i]=0;}
    for(int i=1;i<=m;i++) anc[i]=i, rk[i]=1; tot=m;
}
inline void dfs_pre(int x,int f) {
    fa[x]=f; d[x]=1;
    for(int e=edge[x].size()-1;e>=0;e--) {
        int v=edge[x][e]; if(v==f) continue;
        dfs_pre(v,x); d[x]=max(d[x],d[v]+1);
    } if(d[x]==B || x==1) {d[x]=0; G[x]=1;}
}
inline int ga(int x) {return (anc[x]==x) ? x : ga(anc[x]);}
inline data inc(int x) {
    int xx=ga(g[x].first), yy=ga(g[x].second);
    if(xx==yy) return data(0,0,0);
    if(rk[xx]<rk[yy]) swap(xx,yy);
    anc[yy]=xx; --tot;
    if(rk[xx]==rk[yy]) return ++rk[xx], data(xx,yy,2);
    else return data(xx,yy,1);
}
inline void del(int x) {
    if(!opt[x].type) return;
    rk[opt[x].x]-=(opt[x].type-1);
    anc[opt[x].y]=opt[x].y;
    ++tot;
}
inline int group(int x,int f) {
    static int stk[N], tl; tl=0;
    while(x^f) {
        opt[x]=inc(x); 
        stk[++tl]=x; x=fa[x];
    } int rs=tot;
    for(int i=tl;i>=1;i--) del(stk[i]);
    return rs;
}
inline void dfs2(int x,int f) {
    opt[x]=inc(x);
    if(qry[rt].find(x-n)!=qry[rt].end()) 
        ans[x-n]=group(x-n,rt), qry[rt].erase(qry[rt].find(x-n)); 
    for(int e=edge[x].size()-1;e>=0;e--) {
        int v=edge[x][e]; if(v==f) continue;
        dfs2(v,x);
    } del(x);
}
inline void dfs(int x,int f) {
    opt[x]=inc(x);
    if(G[x]) rt=x, dfs2(n+1,0);
    for(int e=edge[x].size()-1;e>=0;e--) {
        int v=edge[x][e]; if(v==f) continue;
        dfs(v,x);
    } del(x);
}
inline void solve() {
    n=rd(), m=rd(); init();
    for(int i=1;i<=n;i++) 
        g[i]=pii(rd(),rd());
    for(int i=1;i<n;i++) {
        int x=rd(), y=rd();
        edge[x].push_back(y);
        edge[y].push_back(x);
    }
    for(int i=1;i<=n;i++)
        g[i+n]=pii(rd(),rd());
    for(int i=1;i<n;i++) {
        int x=rd()+n, y=rd()+n;
        edge[x].push_back(y);
        edge[y].push_back(x);
    } dfs_pre(1,0); dfs_pre(n+1,0);
    for(int i=1,j;i<=n;i++) {
        for(j=i;!G[j];j=fa[j]);
        qry[j][i]=1;
    } dfs(1,0);
    for(int i=1;i<=n;i++)
        W(ans[i]), putchar('\n');
}

int main() {
    for(int T=rd();T;T--) solve();
}