BZOJ 1015: [JSOI2008]星球大战starwar

倒着用并查集维护即可。

#include <bits/stdc++.h>
using namespace std;
const int N=2e5+5;
int n,m,k,x,y,xx,yy,total;
//注意：有2m个点，所以f,broken,head的范围都要开到2m
//k的范围没有说明，但是ans最好开到2m 
int f[N<<1],u[N],v[N],a[N],broken[N<<1],ans[N<<1];
int cnt,head[N<<1];
struct egde{int next,to;}e[N<<1];

int find(int x)
{
	if (f[x]==x) return x;
	return f[x]=find(f[x]);
}

inline void add(int u,int v)
{
	cnt++;
	e[cnt].next=head[u];
	e[cnt].to=v;
	head[u]=cnt;
}

int main(){
	//简化题意：给一幅图，k个删除操作，按顺序删除给定的每一个点后，求图中的联通块个数
	
	//可以倒着做，先求出把k个点都删除后有多少个联通块个数，存在ans[k]中
	//然后，就可以把第k个点当做是没有删除的点了，因为对于k-1的时候，k这个点没有被删除
	//我们可以知道，有了k这个点，那么联通块个数可能会减少一点，至于是减多少？
	//就是 ： 枚举所有和a[k]相连的点，如果这些和a[k]相连的点不是被删除的点且它没有和a[k]在一个联通块内
	//那么就可以把 联通块个数减一，然后把这个点和a[k]利用并查集连接起来
	//注意一下，我们发现，如果上面所说的可以使得联通块个数减一的点只有一个，然后联通块个数减了一
	//其实是错误的算法，因为我们会发现这个点和a[k]相连了以后，联通块个数不变啊
	//哦，原来，把a[k]恢复时，a[k]自己就算是一个联通块了，所以每次操作前total++
	//对于这个步骤，之前我没注意到。不过，只要能想到倒着做的话,对于这个步骤是很容易调的 
	scanf("%d%d",&n,&m);
	for (register int i=1; i<=n; ++i) f[i]=i;
	for (register int i=1; i<=m; ++i) scanf("%d%d",&u[i],&v[i]),u[i]++,v[i]++;
	
	scanf("%d",&k);
	for (register int i=1; i<=k; ++i) scanf("%d",&a[i]),a[i]++,broken[a[i]]=1; 
	for (register int i=1; i<=m; ++i)
	if (!broken[u[i]] && !broken[v[i]])
	{
		x=u[i]; y=v[i];
		xx=find(x); yy=find(y);
		f[xx]=yy;
	}
	for (register int i=1; i<=n; ++i) if (!broken[i] && find(i)==i) total++;
	ans[k]=total;  //全部点都被入侵后的联通块个数 
	
	for (register int i=1; i<=m; ++i)
	{
		add(u[i],v[i]);
		add(v[i],u[i]);	
	}
	
	for (register int i=k-1; i>=0; --i)
	{
		broken[a[i+1]]=0;
		for (register int j=head[a[i+1]]; j; j=e[j].next)
		{
			x=a[i+1]; y=e[j].to;
			xx=find(x); yy=find(y);
			if (!broken[x] && !broken[y] && xx!=yy)
			{
				f[xx]=yy;
				total--;
			}
		}
		total++;
		ans[i]=total;
	}
	for (register int i=0; i<=k; ++i) printf("%d\n",ans[i]);
return 0;
}