连通图（图论）

题目描述:

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

输入格式:

每组数据的第一行是两个整数 n 和 m（0<=n<=1000）。n 表示图的顶点数目，m 表示图中边的数目。如果 n 为 0 表示输入结束。随后有 m 行数据，每行有两个值 x 和 y（0<x, y <=n），表示顶点 x 和 y 相连，顶点的编号从 1 开始计算。输入不保证这些边是否重复。

输出格式:

对于每组输入数据，如果所有顶点都是连通的，输出"YES"，否则输出"NO"。

样例输入:

4 3
4 3
1 2
1 3
5 7
3 5
2 3
1 3
3 2
2 5
3 4
4 1
7 3
6 2
3 1
5 6
0 0

样例输出:

YES
YES
NO

代码如下：

#include <iostream>
using namespace std;
int a[1001]={0};
int v,e;
void f(int v1, int v2) {
	int a1=a[v1];
	int a2=a[v2];
	if(a1==a2){
		return;
	}else{
		a[v1]=a[v2];
		for(int i=1;i<=v;i++) {
			if(a[i]==a1){
				a[i]=a[v2];
			}	
		}
	}
}
int main(){   
	while(1){
		cin>>v>>e;
		if (v==0&&e==0){
			break;
		}
		for(int i=1;i<=v;i++){
			a[i]=i;
		}
		int v1,v2;
		for(int i=1;i<=e;i++){
			cin>>v1>>v2;
			f(v1,v2);
		}
		int k=0;
		for(int i=1;i<=v;i++){
			if(a[i]==i){
				k++;
			}
		}
		if(k==1){
			cout<<"YES"<<endl;
		}else{
			cout<<"NO"<<endl;	
		}
	}
	return 0;
}