Problem Description
为了训练小希的方向感,Gardon建立了一座大城堡,里面有N个房间(N<=10000)和M条通道(M<=100000),每个通道都是单向的,就是说若称某通道连通了A房间和B房间,只说明可以通过这个通道由A房间到达B房间,但并不说明通过它可以由B房间到达A房间。Gardon需要请你写个程序确认一下是否任意两个房间都是相互连通的,即:对于任意的i和j,至少存在一条路径可以从房间i到房间j,也存在一条路径可以从房间j到房间i。
Input
输入包含多组数据,输入的第一行有两个数:N和M,接下来的M行每行有两个数a和b,表示了一条通道可以从A房间来到B房间。文件最后以两个0结束。
Output
对于输入的每组数据,如果任意两个房间都是相互连接的,输出"Yes",否则输出"No"。
Sample Input
3 3 1 2 2 3 3 1 3 3 1 2 2 3 3 2 0 0
Sample Output
Yes No
题目的意思就是找到任意两个房间是否连通。
循环找到房间1到房间n可以到达的房间,由于到达别的房间可以不只一步,因此这里需要用到dfs搜索。
这道题使用set容器来储存每个房间可到达的房间编号。
AC代码如下:
//
// main.cpp
// 1269.1
//
// Created by showlo on 2018/5/16.
// Copyright © 2018年 showlo. All rights reserved.
//
#include <iostream>
#include <vector>
#include <set>
#define mvs memset(vis,0,sizeof(vis))
using namespace std;
int m,n;
set<int> s[10001];
int vis[10001];
void dfs(int ori,int x){
set<int>::iterator it;
vector<int> v;
for(it=s[x].begin();it!=s[x].end();it++)
v.push_back(*it);
vector<int>::iterator vit;
for (vit=v.begin(); vit!=v.end(); vit++) {
if(vis[*vit]==1)
continue;
s[ori].insert(*vit);
vis[*vit]=1;
dfs(ori,*vit);
}
}
int main() {
while(cin>>n>>m&&(n||m)) //n个房间、m个通道
{
while(m--)
{
int a,b;
cin>>a>>b;
s[a].insert(b);
}
bool flag=1;
for(int i=1;i<=n;i++){
if(s[i].size()==n-1)
continue;
mvs;
vis[i]=1;
dfs(i,i);
if(s[i].size()!=n-1)
{
flag=0;
break;
}
}
if (flag==1) {
cout<<"Yes"<<endl;
}
else
cout<<"No"<<endl;
for(int i=1;i<=n;i++)
s[i].clear();
}
return 0;
}