Problem Description
某省调查城镇交通状况,得到现有城镇道路统计表,表中列出了每条道路直接连通的城镇。省政府“畅通工程”的目标是使全省任何两个城镇间都可以实现交通(但不一定有直接的道路相连,只要互相间接通过道路可达即可)。问最少还需要建设多少条道路?
Input
测试输入包含若干测试用例。每个测试用例的第1行给出两个正整数,分别是城镇数目N ( < 1000 )和道路数目M;随后的M行对应M条道路,每行给出一对正整数,分别是该条道路直接连通的两个城镇的编号。为简单起见,城镇从1到N编号。
注意:两个城市之间可以有多条道路相通,也就是说
3 3
1 2
1 2
2 1
这种输入也是合法的
当N为0时,输入结束,该用例不被处理。
Output
对每个测试用例,在1行里输出最少还需要建设的道路数目。
Sample Input
4 2 1 3 4 3 3 3 1 2 1 3 2 3 5 2 1 2 3 5 999 0 0
Sample Output
1 0 2 998
Hint
Hint Huge input, scanf is recommended.
Source
利用并查集解题。
这里要特别注意,init初始化时要初始化n + 1个点,因为0号顶点不存在,而n号顶点却存在。
思路:
建图。将城市看作点,城市与城市之间联通代表这对顶点之间有一条边相连。
m次循环,将联通的两点加入集合中。然后对每对顶点查询是否在同一个集合中,如果不在,则计数器cnt++,表示要在原来的基础上增加一条道路;同时,必须紧接着将这一对原本不在同一集合中的顶点加入到同一集合中即unite(i, j)。
最后输出cnt即可。
这里的输入还要注意以下,先读入n,判断n是否为0,若n为0,则直接退出while循环或者直接return 0
C++代码如下:
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int MAX_N = 1010;
const int MAX_M = MAX_N * (MAX_N + 1) / 2;
int par[MAX_N]; // 父亲
int Rank[MAX_N]; // 树的高度
// 初始化n个元素
void init(int n) // 一定不要忘了初始化!!!
{
for (int i = 0; i < n; i++)
{
par[i] = i; // 初始化i的父亲:一开始,每个节点都是自己的祖先
Rank[i] = 0; // 初始化i的高度,一开始,每个节点都是自己的祖先即树根,而树根的高度为0
}
}
// 查询并返回树的根
int find(int x)
{
if (par[x] == x)
return x;
else
return par[x] = find(par[x]); // return a = b; 的意思是,将b的值赋值给a,然后返回a的值
// 这里的意思是将x的父亲的父亲赋值给x的父亲,也就是已知沿着树干向上找根节点
}
// 合并x和y所在的集合
void unite(int x, int y)
{
x = find(x); y = find(y); // 将x和y的根分别赋值给x和y
if (x == y) return; // 如果x于y的根相同,则不必合并
if (Rank[x] < Rank[y]) /// 合并时如果两棵树的高度不同,则将Rank小的连到Rank到的上面
par[x] = y;
else
{
par[y] = x;
if (Rank[x] == Rank[y]) Rank[x]++; // 如果相同,则最高的高度合并之后需要在原来的基础上+1
}
}
bool same(int x, int y) // 判断x和y是否属于同一个集合
{
return find(x) == find(y); // 只需判断x和y的根节点是否相同
}
int main()
{
int n, m;
int a, b;
while (scanf("%d", &n))
{
if (n == 0)
break;
int cnt = 0;
scanf ("%d", &m);
init(n + 1);
for (int i = 0; i < m; i++)
{
scanf ("%d%d", &a, &b);
unite(a, b); // 将道路分别加入到某些集合中
}
for (int i = 1; i < n; i++)
{
for (int j = i + 1; j <= n; j++)
{
if (!same(i, j))
{
unite(i, j);
cnt++;
}
}
}
printf ("%d\n", cnt);
}
return 0;
}