7-53 生化危机——dfs

人类正在经历一场生化危机，许多城市已经被病毒侵袭，这些城市中的人们为了避免感染病毒，计划开车逃往其他没有被病毒入侵的城市（安全城市）。有些城市之间有公路直达，有些没有。虽然他们知道哪些城市是安全的，但是不知道有没有一条安全路径能够到达安全城市（只有该路径上经过的所有城市都是安全的，该路径才是安全路径）。请你编写一个程序帮助他们判断。

输入格式:
输入第一行为三个正整数，分别表示所有城市个数m（m<=100）、安全城市个数n（m<=50）、公路个数k（k<=100）。随后一行给出n个安全城市的编号。随后k行，每一行给出两个整数，表示连接一条公路的两个城市编号。最后一行输入两个整数，分别表示当前所在城市s、目标城市d。每行整数之间都用空格分隔。

输出格式:
若目标城市已被病毒入侵（非安全城市），输出"The city i is not safe!“；若目标城市为安全城市且从当前所在城市能够经过一条安全路径到达目标城市，输出"The city can arrive safely！”；若目标城市为安全城市但是从当前所在城市没有一条安全路径到达目标城市，输出"The city can not arrive safely！"，i为目标城市编号。

输入样例1:
5 2 5
3 4
0 1
0 2
0 4
1 2
2 4
0 4
输出样例1:
The city 4 can arrive safely!
输入样例2:
5 2 5
3 4
0 1
0 2
0 4
1 2
2 4
0 3
输出样例2:
The city 3 can not arrive safely!
输入样例3:
5 2 5
3 4
0 1
0 2
0 4
1 2
2 4
0 1
输出样例3:
The city 1 is not safe!

分析

就是比较简单的dfs，属于图上的dfs，爆搜起点能到达的地方，如果x到达的城市安全且有路，从这个中转点x再往下搜，至到搜到终点，否则输出相应的结果；印象第一次做这个题是去年吧，当时还是一脸懵，时间过得可真快…

#include<bits/stdc++.h>

using namespace std;

int m, n, k, s, d, flag;
int safe[105];//标记安全城市
int path[105][105];

void dfs(int x) {
    if (x == d) {
        flag = 1;
        return;
    }
    //看x能到达的地方
    for (int i = 0; i < n; ++i) {
        if (path[x][i] && safe[i]) {
            dfs(i);
        }
    }
}

int main() {
    cin >> n >> m >> k;
    for (int i = 1; i <= m; ++i) {
        int t;
        cin >> t;
        safe[t] = 1;
    }
    int x, y;
    for (int i = 0; i < k; ++i) {
        cin >> x >> y;
        path[x][y] = 1;
        path[y][x] = 1;
    }
    cin >> s >> d;
    if (!safe[d]) {
        printf("The city %d is not safe!", d);
        return 0;
    }
    dfs(s);
    if (flag) {
        printf("The city %d can arrive safely!", d);
    } else {
        printf("The city %d can not arrive safely!", d);
    }
    return 0;
}