图结构练习——BFSDFS——判断可达性

Problem Description

 在古老的魔兽传说中，有两个军团，一个叫天灾，一个叫近卫。在他们所在的地域，有n个隘口，编号为1..n，某些隘口之间是有通道连接的。其中近卫军团在1号隘口，天灾军团在n号隘口。某一天，天灾军团的领袖巫妖王决定派兵攻打近卫军团，天灾军团的部队如此庞大，甚至可以填江过河。但是巫妖王不想付出不必要的代价，他想知道在不修建任何通道的前提下，部队是否可以通过隘口及其相关通道到达近卫军团展开攻击。由于n的值比较大（n<=1000），于是巫妖王找到了擅长编程的你 =_=，请你帮他解决这个问题，否则就把你吃掉变成他的魔法。为了拯救自己，赶紧想办法吧。

 

Input

 输入包含多组，每组格式如下。

第一行包含两个整数n,m（分别代表n个隘口，这些隘口之间有m个通道）。

下面m行每行包含两个整数a，b；表示从a出发有一条通道到达b隘口（注意：通道是单向的）。

Output

 如果天灾军团可以不修建任何通道就到达1号隘口，那么输出YES，否则输出NO。

 

Example Input

2 1
1 2
2 1
2 1

Example Output

NO
YES

code:

#include<iostream>
#include<cstring>
using namespace std;
int map[1010][1010], visit[1010], count[1010];
int len, n, m;
void dfs(int x);
int main()
{
    int a, b;
    int i;
    while(cin>>n>>m)
    {
        memset(map, 0, sizeof(map));
        memset(visit, 0, sizeof(visit));
        for(i = 1;i<=m;i++)
        {
            cin>>a>>b;
            map[a][b] = 1;
        }
        len = 0;
        dfs(n);
        for(i = 0;i<len;i++)
        {
            if(count[i]==1) break;
        }
        if(i==len) cout<<"NO"<<endl;
        else cout<<"YES"<<endl;
    }
}
void dfs(int x)
{
    int i;
    visit[x] = 1;
    count[len++] = x;//利用递归，找出是否有指向编号1的路径
    for(i = 1;i<=n;i++)
    {
        if(!visit[i]&&map[x][i])
        {
            dfs(i);
        }
    }
}