DTOJ 4710. 并

题意

一棵 $n$ 个点的树有 $n-1$ 条边，分别为 $(u_1,v_1),(u_2,v_2),\cdots ,(u_{n-1},v_{n-1})$. 开始时，对每个点 $u$，有集合 S u = { u } S_u=\{u\} Su​={u}.

有 $m$ 个操作，第 $i$ 次操作给定 $p_i$，将 $S_{u_{p_i}}$ 与 $S_{v_{p_i}}$ 赋为它们的并。

所有操作结束后，对所有的 $i\in [1,n]$，求包含 $i$ 的集合的数目。

#1 ($30$pts), $n,m\le 100$.

#2 ($30$pts), $n,m\le 1000$.

#3 ($40$pts), $n,m\le 5\times 10^5$.

题解

发现要求的是每个点按照时间顺序沿着给定的边延伸后，能到达的集合大小。直接处理这个问题不太容易，注意到边是双向的，先反过来考虑：能否计算能到达$u$的点集的大小？考虑每次练一条边$(u,v)$，$u,v$的大小变为$z[u]+z[v]-z[u\&v]$，而$z[u\&v]$恰好等于上一次连接$(u,v)$后$u,v$的大小！原来的问题是这个的逆过程，反过来做即可。