题目描述
给定一个无向图和其中的所有边,判断这个图是否所有顶点都是连通的。
输入
每组数据的第一行是两个整数 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,以及只有一个点的情况。