题目描述:
给定一个无向图和其中的所有边,判断这个图是否所有顶点都是连通的。
输入:
每组数据的第一行是两个整数 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 <iostream>
#include<cstdio>
using namespace std;
const int MAXN=100;
int father[MAXN];
int height[MAXN];
//初始化
void initial(int n){
for(int i=1;i<=n;i++){//一般情况下这里是从0开始,小于n结束,此题例外。
father[i]=i;
}
}
//查找某个节点,返回它坐在集合的根节点
int Find(int n){
if(n==father[n]){
return n;
}
father[n]=Find(father[n]);
return father[n];
}
//求并集
void Union(int m,int n){
//求出m和n所在的集合,即返回他们所在集合的根节点
m=Find[m];
n=Find[n];
if(m==n){
return;
}
else if(height[m]<height[n]){
father[m]=n;
}
else if(height[m]>height[n]){
father[n]=m;
}
else{
father[n]=m;
height[m]++;
}
}
int main()
{int n,m;
while(scanf("%d%d",&n,&m)!=EOF){
if(n==0){
break;
}
initial(n);
while(m--){
int a;
int b;
scanf("%d%d",&a,&b);
Union(a,b);
}
int answer=0;
for(int i=1;i<=n;i++){
if(i==father[i]){
answer++;
}
}
if(answer==1){
printf("YES\n");
}
else{
printf("NO\n");
}
}
return 0;
}
小结:这里顺便贴上了并查集中的find操作,以及initial初始化的代码,具有一般通用性。