HDU 1269 迷宫城堡(DFS)

时间:2021-04-11 06:58:10

迷宫城堡

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[];
set<int> s[];
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]=;
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=;
for(int i=; i<=n; i++)
{
if((int)s[i].size()==n-)continue;
msv;
vis[i]=;
dfs(i,i);
//剪枝,如果某个房间不能走到其他所有房间,
//就没有继续搜索的意义了
if((int)s[i].size()!=n-)
{
flag=;
break;
}
}
for(int i=;i<=n&&flag;i++)
{
if((int)s[i].size()!=n-)
{
flag=;
}
}
if(flag)printf("Yes\n");
else printf("No\n");
for(int i=;i<=n;i++)
s[i].clear();
}
return ;
}