Luogu P1197 [JSOI2008]星球大战

题面

分析

并查集即可，注意由于并查集只能维护集合求并，不能维护集合“相减”，故需要按照摧毁顺序倒着运行，变减为加，这基本上算是并查集的一个较高级的用法了

code

#include<bits/stdc++.h>
using namespace std;
#define loop(i,start,end) for(int i=start;i<=end;i++)
#define anti_loop(i,start,end) for(int i=start;i>=end;i--)
#define clean(arry,num); memset(arry,num,sizeof(arry));
int n,m,k,num,cnt=0;
const int maxn=400000+10,maxm=200000+5;
int fa[maxn];
int con[maxm][2];
int att[maxn];
bool bro[maxn];
stack<int>ans;
struct node{int e;int nxt;}edge[maxm<<1];
int head[maxn];
inline int read()
{
   int ans=0;
   char r=getchar();bool neg=false;
   while(r<'0'||r>'9'){if(r=='-')neg=true;r=getchar();}
   while(r>='0'&&r<='9'){ans=ans*10+r-'0';r=getchar();}
   return (neg)?-ans:ans;
}
int findfa(int a)
{
	return fa[a]=(fa[a]==a)?fa[a]:findfa(fa[a]);
}
inline void addl(int u,int v)
{
	edge[cnt].e=v;
	edge[cnt].nxt=head[u];
	head[u]=cnt++;
}
int main()
{
	freopen("data.txt","r",stdin);
	n=read();m=read();clean(bro,false);num=n;
	loop(i,0,n-1)
	{
		fa[i]=i;
		head[i]=-1;
	}
	loop(i,0,m-1)
	{	con[i][0]=read();
		con[i][1]=read();
		addl(con[i][0],con[i][1]);
		addl(con[i][1],con[i][0]);
	}
	k=read();
	num-=k;
	loop(i,0,k-1)
	{
		scanf("%d",&att[i]);
		bro[att[i]]=true;
	}
	loop(i,0,m-1)
	{
		int x=con[i][0];int y=con[i][1];
		if(!bro[x]&&!bro[y])
		{
			if(findfa(x)!=findfa(y))num--;
			fa[findfa(x)]=findfa(y);	
		}
	}
	//初始状态值 
	ans.push(num);//利用栈先进后出的特性来保存答案
	anti_loop(i,k-1,0)//倒序更新
	{
		int u=att[i];
		bro[u]=false;
		num++;//由于新恢复了一个点，连通块的个数增加一个
		for(int i=head[u];i!=-1;i=edge[i].nxt)
		{
			int v=edge[i].e;
			if(!bro[v]&&findfa(v)!=findfa(u))//如果与之相连的点没有被攻击且之前未连接过，联通块个数减少
			{
				fa[findfa(v)]=findfa(u);
				num--;
			}
		ans.push(num);	
	}
	while(ans.empty()==false)
	{
		printf("%d\n",ans.top());
		ans.pop();
	}
	return 0;
}

学到的东西

• 回顾并查集

数据支持：fa数组，fa[i]表示i的父节点
函数支持：findfa(i)函数，返回i的父亲;sum(i,j)函数，将i和j所在集合合并；query(i,j)函数，查询i和j是否在同一个集合内