并查集进阶版


过关代码如下

cpp 复制代码
#define _CRT_SECURE_NO_WARNINGS
#include<bits/stdc++.h>
#include<unordered_set>
using namespace std;

int n, m;
vector<int> edg[400005];
int a[400005], be[400005]; // a的作用就是存放要摧毁
int k;
int fa[400005];
int daan[400005];

void add(int x, int y) {
	edg[x].push_back(y);
	edg[y].push_back(x);
}

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

void uni(int x, int y) {
	int xx = find(x), yy = find(y);
	if (xx == yy) return;
	fa[xx] = yy;
}

int main() {
	cin >> n >> m;
	int l, r;
	for (int i = 1; i <= m; i++) {
		cin >> l >> r;
		add(l, r); // 建立边
	}
	cin >> k;
	for (int i = 1; i <= k; i++) {
		cin >> a[i];
		be[a[i]] = 1; // 标记为1,表示被摧毁
	}
	int ans = n-k; // 一开始的时候每个点都是一个块
	// 初始化并查集
	for (int i = 0; i <= n; i++) fa[i] = i;
	// 开始区分联通分量
	for (int i = 0; i < n; i++) {
		if (be[i]) continue;
		for (int u : edg[i]) {
			if (be[u]) continue;
			if (find(i) == find(u)) continue;
			uni(i, u);// 连接
			ans--;
		}
	}
	daan[k+1] = ans;
	for (int i = k; i >= 1; i--) {
		//cout << "yes" << endl;
		int xiufu = a[i];
		ans++;
		be[xiufu] = 0;  // 恢复为1
		for (int u : edg[xiufu]) {
			if (be[u]) continue;
			if (find(u) == find(xiufu)) continue;
			ans--;
			uni(u, xiufu);
		}
		daan[i] = ans;
	}
	for (int i = 1; i <= k+1; i++) {
		cout << daan[i] << endl;
	}
	//cout << " now";
	return 0;
}
相关推荐
焜昱错眩..8 小时前
代码随想录算法训练营第三十九天|62.不同路径 63.不同路径ll
算法
焦耳加热11 小时前
阿德莱德大学Nat. Commun.:盐模板策略实现废弃塑料到单原子催化剂的高值转化,推动环境与能源催化应用
人工智能·算法·机器学习·能源·材料工程
wan5555cn11 小时前
多张图片生成视频模型技术深度解析
人工智能·笔记·深度学习·算法·音视频
u60612 小时前
常用排序算法核心知识点梳理
算法·排序
蒋星熠14 小时前
Flutter跨平台工程实践与原理透视:从渲染引擎到高质产物
开发语言·python·算法·flutter·设计模式·性能优化·硬件工程
小欣加油14 小时前
leetcode 面试题01.02判定是否互为字符重排
数据结构·c++·算法·leetcode·职场和发展
3Cloudream14 小时前
LeetCode 003. 无重复字符的最长子串 - 滑动窗口与哈希表详解
算法·leetcode·字符串·双指针·滑动窗口·哈希表·中等
王璐WL15 小时前
【c++】c++第一课:命名空间
数据结构·c++·算法
空白到白15 小时前
机器学习-聚类
人工智能·算法·机器学习·聚类
索迪迈科技15 小时前
java后端工程师进修ing(研一版 || day40)
java·开发语言·学习·算法