并查集学习心得

int find(int x)//并查集找父亲
{
	if(x!=fa[x]) fa[x]=find(fa[x]);
	return fa[x];
}
void add(int x,int y)//合并 
{
	int fx=find(x);
	int fy=find(y);
	if(x!=y) fa[fx]=fy;
}

P1197 [JSOI2008] 星球大战 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

洛谷p1197星球大战 :并查集+逆向思考_phrame_的博客-CSDN博客

这题由于并查集不好进行删除点的操作,所以反向操作,把删除点反向成一个个添加新的点,注意判断连通块的方法是看,fa[i]==i,那么是一个连通块,父亲相等为一个连通块

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N=1e6+100;
vector<int>ans;
vector<int> a[N];//存图
int fa[N];
int h[N];//存要摧毁的 
int vis[N];//标记是否要摧毁 
int find(int x)//并查集 
{
	if(x!=fa[x]) fa[x]=find(fa[x]);
	return fa[x];
}
void add(int x,int y)//合并 
{
	int fx=find(x);
	int fy=find(y);
	if(x!=y) fa[fx]=fy;
}
signed main()
{
	int n,m;
	cin>>n>>m;
	for(int i=0;i<n;i++)
	{
		fa[i]=i;//初始化 
	}
	for(int i=1;i<=m;i++)
	{
		int u,v;
		cin>>u>>v;
		a[u].push_back(v);
		a[v].push_back(u);//双向存图 
	}
	int k;
	cin>>k;
	for(int i=1;i<=k;i++)
	{
		cin>>h[i];
		vis[h[i]]=1;//标记要摧毁的 
	}
	for(int i=0;i<n;i++)
	{
		for(auto x:a[i])//遍历a[i]所有的点 
		{
			if(vis[x] ||vis[i])//去掉要摧毁的 
			{
				continue;
			}
			add(x,i);//对不摧毁的点合并处理 
		}
	}
	int cnt=0;
	for(int i=0;i<n;i++)
	{
		if(!vis[i]&&fa[i]==i)
		{
			cnt++;//没有修复前连通块的个数 
		}
	 } 
	 ans.push_back(cnt);//最后一种全摧毁完的答案
	 //开始修复,倒推
	 for(int i=k;i>=1;i--)
	 {
	 	int x=h[i];
	 	cnt++;//把它当成独立的连通块 
	 	vis[x]=0;//被修复 
	 	for(auto u:a[x])
	 	{
	 		if(!vis[u]&&find(x)!=find(u))
	 		{
	 			add(u,x); 
	 			cnt--;//减去不是连通块的 
			 }
		 }
		 ans.push_back(cnt);
	  } 
	  reverse(ans.begin(),ans.end());
	  for(auto c:ans) 
	  {
	  	cout<<c<<endl;
	  }
	  return 0;
}
相关推荐
little redcap27 分钟前
第十九次CCF计算机软件能力认证-乔乔和牛牛逛超市
数据结构·c++·算法
muyierfly1 小时前
34.贪心算法1
算法·贪心算法
luthane3 小时前
python 实现average mean平均数算法
开发语言·python·算法
静心问道4 小时前
WGAN算法
深度学习·算法·机器学习
杰九4 小时前
【算法题】46. 全排列-力扣(LeetCode)
算法·leetcode·深度优先·剪枝
manba_4 小时前
leetcode-560. 和为 K 的子数组
数据结构·算法·leetcode
liuyang-neu4 小时前
力扣 11.盛最多水的容器
算法·leetcode·职场和发展
忍界英雄4 小时前
LeetCode:2398. 预算内的最多机器人数目 双指针+单调队列,时间复杂度O(n)
算法·leetcode·机器人
Kenneth風车4 小时前
【机器学习(五)】分类和回归任务-AdaBoost算法-Sentosa_DSML社区版
人工智能·算法·低代码·机器学习·数据分析
C7211BA5 小时前
使用knn算法对iris数据集进行分类
算法·分类·数据挖掘