【JZOJ 杂题选讲】Bitset Master

题目

思路

换个问题，求每个集合最后的大小。我们发现，如果将 u,v 合并，那么 f[u]=f[v]=f[u]+f[v]−f[u]⋂f[v]

而 f[u]⋂f[v] 之和上一次 u,v 合并的结果有关，于是我们可以对每条边单独记录一个数，表示上一次合并这条边的结果

回到原问题，我们发现，每个点被哪些集合包含，只需要倒叙处理新问题就可以得到原问题的答案

代码

#include<bits/stdc++.h>
using namespace std;
const int N=1e6+77;
int n,m,p[N],x[N],y[N],f[N],g[N];
void write(int x,int flag)
{
    printf("%d",x);
    if(flag==0) putchar(' ');
    else puts("");
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1; i<n; i++) scanf("%d%d",&x[i],&y[i]);
    for(int i=1; i<=m; i++) scanf("%d",&p[i]);
    for(int i=1; i<=n; i++) f[i]=1;
    for(int i=m; i>=1;--i)
	{
        int u=x[p[i]], v=y[p[i]];
        f[u]=f[v]=f[u]+f[v]-g[p[i]];
        g[p[i]]=f[u];
    }
    for(int i=1; i<=n; i++) write(f[i],i==n);
}