九度OJ题目1109-连通图

题目描述:

给定一个无向图和其中的所有边,判断这个图是否所有顶点都是连通的。

输入:

每组数据的第一行是两个整数 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");
		}
	}
}
发布了53 篇原创文章 · 获赞 3 · 访问量 3505

猜你喜欢

转载自blog.csdn.net/sinat_38292108/article/details/88342825