【洛谷 P1536】村村通 题解（集合+并查集+路径压缩）

村村通

题目描述

某市调查城镇交通状况，得到现有城镇道路统计表。表中列出了每条道路直接连通的城镇。市政府 “村村通工程” 的目标是使全市任何两个城镇间都可以实现交通（但不一定有直接的道路相连，只要相互之间可达即可）。请你计算出最少还需要建设多少条道路？

输入格式

输入包含若干组测试数据，每组测试数据的第一行给出两个用空格隔开的正整数，分别是城镇数目 $n$ 和道路数目 $m$ ；随后的 $m$ 行对应 $m$ 条道路，每行给出一对用空格隔开的正整数，分别是该条道路直接相连的两个城镇的编号。简单起见，城镇从 $1$ 到 $n$ 编号。

注意：两个城市间可以有多条道路相通。

在输入数据的最后，为一行一个整数 $0$，代表测试数据的结尾。

输出格式

对于每组数据，对应一行一个整数。表示最少还需要建设的道路数目。

样例 #1

样例输入 #1

4 2
1 3
4 3
3 3
1 2
1 3
2 3
5 2
1 2
3 5
999 0
0

样例输出 #1

1
0
2
998

提示

数据规模与约定

对于 $100\%$ 的数据，保证 $1\le n<1000$ 。

---

思路

首先，从输入中读取测试用例的数量。每个测试用例包含城市数量 n 和道路数量 m，以及 m 对城市之间的连接关系。

在处理每个测试用例之前，执行 init 函数初始化 pre 数组和 rs 集合。pre 数组用于存储并查集的父节点信息，rs 集合用于存储并查集的根节点集合。

init 函数将每个城市的父节点设为自身，并清空 rs 集合。

然后，读取 m 对城市之间的连接关系。每读取一对城市 u 和 v，就调用 merge 函数将它们合并到同一个集合中。

merge 函数首先通过 root 函数找到 a 和 b 的根节点。如果它们的根节点相同，就不做任何操作。否则，将 a 的根节点的父节点设为 b 的根节点，使 a 和 b 属于同一个集合。

root 函数用于查找元素 a 的根节点。如果 a 的父节点不是 a 本身，就继续查找其父节点的根节点。最后，将 a 的父节点更新为其根节点，以实现路径压缩。

最后，遍历所有城市，将每个城市的根节点插入 rs 集合。由于集合自动去重，rs 集合的大小就是并查集的集合数量。输出 rs 集合的大小减一，即最少还需要建设的道路数量。

---

AC代码

#include <iostream>
#include <algorithm>
#include <set>
#define AUTHOR "HEX9CF"
using namespace std;
using ll = long long;

const int N = 1e6 + 7;
const int INF = 0x3f3f3f3f;
const int MOD = 1e9 + 7;

ll n, m;
ll pre[N];
set<ll> rs;

ll root(ll a) {
	ll i = a;
	while(pre[i] != i) {
		i = pre[i];
	}
	return (pre[a] = i);
}

void merge(ll a, ll b) {
	ll ra = root(a);
	ll rb = root(b);
	if(ra == rb) {
		return;
	}
	pre[ra] = rb;
}

void init() {
	for(int i = 1; i <= n; i++) {
		pre[i] = i;
	}
	rs.clear();
}

int main() {
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	
	while(cin >> n) {
		if(!n) {
			break;
		}
		init();
		cin >> m;
		for(int i = 1; i <= m; i++) {
			ll u, v;
			cin >> u >> v;
			merge(u, v);
		}
		for(int i = 1; i <= n; i++) {
			rs.insert(root(i));
		}
		cout << ((int)rs.size() - 1) << "\n";
	}
	return 0;
}