问题 B: 连通图

题目描述

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

输入

每组数据的第一行是两个整数 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<stdio.h>
#include<set>
#include<string.h>
using namespace std;
const int MAXV=1010;
bool vis[MAXV]={true};
set<int> Adj[MAXV];
void DFS(int u){
	vis[u]=true;
	for(set<int>::iterator i=Adj[u].begin();i!=Adj[u].end();i++){
		int v=*i;
		if(vis[v]==false)
			DFS(v);
	}
}
int DFSTrave(int n){
	int ans=0;
	for(int i=1;i<=n;i++)
	{
		if(vis[i]==false)
		{
			ans++;
			DFS(i);
		}
	}
	return ans;
}
int main(){
	int a,b,n,m;
	while(scanf("%d %d",&n,&m)!=EOF&&n!=0){
		for(int i=1;i<MAXV;i++){
			vis[i]=true;
			Adj[i].clear();
		}	
		int x=m;
		for(int i=1;i<=n;i++)vis[i]=false;	
		while(x--){
			scanf("%d %d",&a,&b);
			//if(a!=b){
			Adj[a].insert(b);
			Adj[b].insert(a);	
		//	}
			
		}
		/*for(set<int>::iterator it=Adj[1].begin();it!=Adj[1].end();it++){
			printf("it=%d\n",*it);
		}*/	
	int flag=DFSTrave(n);
	if(flag==1||(n==1&&m==0))printf("YES\n");//printf("flag=%d YES\n",flag);
	else printf("NO\n");//printf("flag=%d NO\n",flag);
	}
	return 0;
}

注：避免加入重合元素用set，以及只有一个点的情况。