HDU 1269 迷宫城堡(DFS)

迷宫城堡

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

 

Answer

暴力DFS，从第一个房间开始搜：1->2->3，因为有环的存在，要判断是否回到了起点，否则会无限循环。

某个房间能走到某个房间用set来记录，不会重复。

 

#include <set>
#include <cmath>
#include <queue>
#include <vector>
#include <cstdio>
#include <string>
#include <cstring>
#include <iostream>
#include <algorithm>
#define ms(a) memset(a,0,sizeof(a))
#define msp memset(mp,0,sizeof(mp))
#define msv memset(vis,0,sizeof(vis))
using namespace std;
int n,m;
bool vis[10001];
set<int> s[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 itv;
    for(itv=v.begin();itv!=v.end();itv++)
    {
        if(vis[*itv])continue;
        //如果要搜索的房间比起点房间编号小，说明那个房间已经搜索过了
        //这时，只要把那个房间的所有能走到的房间insert一遍就行了
        /*//这个剪枝有问题
        if(*itv<ori)
        {
            for(it=s[*itv].begin();it!=s[*itv].end();it++)
            {
                if(*it==ori)continue;
                s[ori].insert(*it);
            }
            continue;
        }
        */
        s[ori].insert(*itv);
        vis[*itv]=1;
        dfs(ori,*itv);
    }
}
int main()
{
    while(cin>>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((int)s[i].size()==n-1)continue;
            msv;
            vis[i]=1;
            dfs(i,i);
            //剪枝，如果某个房间不能走到其他所有房间，
            //就没有继续搜索的意义了
            if((int)s[i].size()!=n-1)
            {
                flag=0;
                break;
            }
        }
        for(int i=1;i<=n&&flag;i++)
        {
            if((int)s[i].size()!=n-1)
            {
                flag=0;
            }
        }
        if(flag)printf("Yes
");
        else printf("No
");
        for(int i=1;i<=n;i++)
            s[i].clear();
    }
    return 0;
}