【HDU】1269 迷宫城堡（C++）

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;
}