洛谷p1197星球大战 :并查集+逆向思考

题目来源:星球大战

解题思路: 并查集+逆向思维

1.首先题目本质上就是一个删点求联通块个数的问题,但是并查集擅长的是合并而不是删除.但是我们反过来想,如果我们现在有的是一个已经进行了删边操作的图(下面我叫它为已删图),然后我们把边加回去,当成一个修复的过程,就可以用并查集来维护啦.

2.我们先用邻接表保存原本的无向图,由于我们先要得到已删图,所以我们只需要先将那些不会被删掉的星球合并到一起,定义一个vis数组标记要被删除的星球即可.

3.然后我们该怎么维护连通块的个数呢?  首先对于已删图的连通块个数(下面叫cnt)

<1>我们可以在合并完已删图的并查集之后遍历所有点 ,,统计满足fa[i] == i && !vis[i]的个数就是连通块的个数(fa[i]是因为在每一个集合中只有根结点的父节点等于它自己,!vis[i]是判断该点属不属于已删图),代码就可以这样写

int cnt = 0; // 未修复之前连通块的个数
	for (int i = 0;i < n; i++) {
		if (!vis[i] && fa[i] == i) cnt++; // 未修复前连通块的个数
	}

<2>我们可以在求已删图的并查集的过程中求连通块个数,最开始每个点的根结点节点都是自己,相当于每一个点都是一个连通块,每合并两个点,连通块的数量就要减一.对于该题有n个星球,减去k个要修复(删除)的星球就是n-k,然后我们进行合并操作,当合并两个根结点不同的星球时就cnt--,最后合并完成也得到了连通块的个数,代码就可以这样写

	int cnt = n - k;
	for (int i = 0;i < n; i++) {
		for (auto x:a[i]) {
			if (vis[x] || vis[i]) continue;
			if (find(x) != find(i)) { // 不属于同一个集合
			    add(x,i);
			    cnt--;
			}
		}
	}

3.然后就是加边(修复)过程,每修复一个星球,我们首先要取消对该星球的标记,然后把它当成一个独立的连通块cnt++,再和并到已删图的并查集中,当合并两个根结点不同的星球时就cnt--,每操作一次把答案存入数组即可,最后逆序输出数组即为答案.

	for (int i = k;i >= 1; i--) {
		int x = h[i];
		cnt++; // 先把x看成一个独立的连通块
		vis[x] = 0; // 被修复
		for (auto u:a[x]) {
			if (!vis[u] && find(x) != find(u)) {
				add(u,x); cnt--;
			}
		} 
		ans.push_back(cnt);
	}

题目的注意点和坑点:

1.题目中的点是从0到n-1,而不是1到n

2.fa数组的初始化最好在所有操作前面

3.注意对逆向思考的理解

ACcode

(其中也有一些我写题时的思路)

// 思路:摧毁星球是删除操作,如果把摧毁看成从最后一个
//      被摧毁的星球开始建立,就变成了加边操作.
//      用并查集就可实现,开始进行合并操作只合并
//      最后不会被摧毁的星球(就是帝国军攻击完之后的状态)
//      然后将原来要删除的星球不断补回去(最终就是帝国军未攻打之前的状态)
//      记录每个状态连通块的个数,最后逆序输出答案即可
// 注意1:并查集fa数组的初始化一定要在任何操作前进行
// 注意2:该题中点是从0到n-1,而不是1到n,看清题目
// 注意3:如果你要在并查集合并过程中求cnt要另开数组存有向边
//       如果用原无向图去遍历会导致cnt的重复计算,因为u-v,v-u会合并两次
#include <bits/stdc++.h>
using namespace std;

using ll = long long ;

const int N = 1e6 + 50;
vector<int> ans; // 存答案
vector<int> a[N]; // 存图
int h[N]; // 存要摧毁(修复)的星球
int fa[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;
}

int 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]) {
			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++; // 先把x看成一个独立的连通块
		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 << '\n';
	return 0;
}