题目描述:
给定一个无向图和其中的所有边,判断这个图是否所有顶点都是连通的。
每组数据的第一行是两个整数 n 和 m(0<=n<=1000)。
n 表示图的顶点数目,m 表示图中边的数目。
如果 n 为 0 表示输入结束。
随后有 m 行数据,每行有两个值 x 和 y(0<x, y <=n),
表示顶点 x 和 y 相连,顶点的编号从 1 开始计算。
输入不保证这些边是否重复。
输出:
对于每组输入数据,如果所有顶点都是连通的,输出”YES”,否则输出”NO”。
4 3
1 2
2 3
3 2
3 2
1 2
2 3
0 0
NO
YES
参考代码:
/*使用并查集,统计连通分量的个数*/
#include<cstdio>
#include<iostream>
using namespace std;
#define N 1001
int Tree[N];
int findRoot(int x) {
if (Tree[x] == -1) return x;
else {
int tmp = findRoot(Tree[x]);
Tree[x] = tmp;
return tmp;
}
}
int main() {
int n, m;
while (scanf("%d%d", &n, &m) != EOF) {
if (n == 0) break;
memset(Tree, -1, sizeof Tree);
//for (int i = 1; i <= n; i++) Tree[i] == -1;
for (int i = 1; i <= m; i++) {
int a, b;
scanf("%d%d", &a, &b);
a = findRoot(a);
b = findRoot(b);
if (a != b) {
Tree[a] = b;
}
}
int cnt = 0;
for (int i = 1; i <= n; i++) {
if (Tree[i] == -1) cnt++;
}
if (cnt == 1) {
printf("YES\n");
}
else {
printf("NO\n");
}
}
}