dfs作为暴力搜索的一种,固然有着相对应的模板,在此我把我的模板分享给大家,这里我以一道例题的形式把让大家理解迷宫问题的dfs写法以及会用dfs写法
迷宫问题
小明置身于一个迷宫,请你帮小明找出从起点到终点的最短路程。
小明只能向上下左右四个方向移动。
输入格式:
第一行是两个整数n和m (1≤ m, n ≤100),表示迷宫的长和宽。接下来是n行,每行m个数字,表示整个迷宫。空地格子用0表示,障碍物用1表示,小明所在起点用3表示,终点用4表示。
输出格式:
如果能够到达终点,输出一个整数,表示小明从起点到目的地所需的最短时间。如果不能到达终点,输出“unreachable”。
输入样例1:
5 5
1 0 1 1 1
1 0 4 1 0
1 0 0 1 0
0 0 0 1 0
1 0 3 0 1
输出样例1:
3
输入样例2:
5 5
3 0 1 1 1
1 0 1 1 0
1 0 1 1 0
0 0 0 1 0
1 0 1 0 4
输出样例2:
unreachable
以下是AC代码(带注释理解)
#include<bits/stdc++.h>
using namespace std;
//G为存储迷宫的图, visit用于判断此路径是否走过(true为走过, false为未走过)
vector<vector<int>>G;
vector<vector<bool>>visit;
//存储最终最短路径所需要的步数
int res = INT_MAX;
//方向数组
vector<pair<int, int>>dirs = { {-1, 0}, {0, 1}, {1, 0}, {0, -1} };
//深度优先搜索
void dfs(int x, int y, int p, int q, int step) {
if (x == p && y == q) {
res = min(res, step);
return;
}
int new_x, new_y;
//这里将以(x, y)为中心的周围四个区域依次进行dfs遍历
for (auto& dir : dirs) {
new_x = x + dir.first;
new_y = y + dir.second;
//如果该点在图的范围内
if (new_x >= 0 && new_x < G.size() && new_y >= 0 && new_y < G[0].size()) {
//如果此刻走的是为走过的路程以及这一块是空地或终点则可进行dfs递归调用
if ((G[new_x][new_y] == 4 || G[new_x][new_y] == 0) && visit[new_x][new_y] == false) {
//走到这里设置为走过
visit[new_x][new_y] = true;
dfs(new_x, new_y, p, q, step + 1);
//走完这里记得设置为未走过
visit[new_x][new_y] = false;
}
}
}
}
int main()
{
int i, j, n, m, start_x, start_y, p, q;
cin >> m >> n;
G.resize(m, vector<int>(n, 0));
visit.resize(m, vector<bool>(n, false));
for (i = 0; i < m; i++) {
for (j = 0; j < n; j++) {
cin >> G[i][j];
if (G[i][j] == 3) {
start_x = i;
start_y = j;
}
if (G[i][j] == 4) {
p = i;
q = j;
}
}
}
//调用dfs之前优先把起点设为已走过
visit[start_x][start_y] = true;
dfs(start_x, start_y, p, q, 0);
if (res == INT_MAX)
cout << "unreachable" << endl;
else
cout << res << endl;
return 0;
}