目录
牛客HJ43 迷宫问题(dfs+回溯)
描述
定义一个二维数组 N*M ,如 5 × 5 数组下所示:
int maze[5][5] = {
0, 1, 0, 0, 0,
0, 1, 1, 1, 0,
0, 0, 0, 0, 0,
0, 1, 1, 1, 0,
0, 0, 0, 1, 0,
};
它表示一个迷宫,其中的1表示墙壁,0表示可以走的路,只能横着走或竖着走,不能斜着走,要求编程序找出从左上角到右下角的路线。入口点为[0,0],既第一格是可以走的路。
数据范围: 2≤n,m≤10 2≤n,m≤10 , 输入的内容只包含 0≤val≤1 0≤val≤1
输入描述:
输入两个整数,分别表示二维数组的行数,列数。再输入相应的数组,其中的1表示墙壁,0表示可以走的路。数据保证有唯一解,不考虑有多解的情况,即迷宫只有一条通道。
输出描述:
左上角到右下角的最短路径,格式如样例所示。
示例1
输入:
5 5 0 1 0 0 0 0 1 1 1 0 0 0 0 0 0 0 1 1 1 0 0 0 0 1 0
输出:
(0,0) (1,0) (2,0) (2,1) (2,2) (2,3) (2,4) (3,4) (4,4)
示例2
输入:
5 5 0 1 0 0 0 0 1 0 1 0 0 0 0 0 1 0 1 1 1 0 0 0 0 0 0
输出:
(0,0) (1,0) (2,0) (3,0) (4,0) (4,1) (4,2) (4,3) (4,4)
注意:不能斜着走!!
解析代码
用回溯法求解,具体步骤为:
- 首先将当前点加入路径,并设置为已走。
- 判断当前点是否为出口,是则输出路径,保存结果;跳转到4。
- 依次判断当前点的上、下、左、右四个点是否可走,如果可走则递归走该点。
- 当前点推出路径,设置为可走。
#include <iostream>
#include <vector>
#include <queue>
using namespace std;
vector<pair<int,int>> path, res;
bool vis[13][13];
int m, n;
int dx[4] = {0, 0, -1, 1};
int dy[4] = {1, -1, 0, 0};
void dfs(vector<vector<int>>& vv, int sr, int sc)
{
path.push_back({sr, sc});
vis[sr][sc] = true;
if(sr == m - 1 && sc == n - 1)
{
res = path; // 一定要保存一下结果
return;
}
queue<pair<int, int>> q;
for(int j = 0; j < 4; ++j)
{
int x = sr + dx[j], y = sc + dy[j];
if(x >= 0 && x < m && y >= 0 && y < n && !vis[x][y] && vv[x][y] == 0)
{
dfs(vv, x, y);
}
}
path.pop_back(); // 回溯
vis[sr][sc] = false;
}
int main()
{
cin >> m >> n;
vector<vector<int>> vv(m, vector<int>(n, 0));
for(int i = 0; i < m; ++i)
{
for(int j = 0; j < n; ++j)
{
cin >> vv[i][j];
}
}
dfs(vv, 0, 0);
for(auto [a, b] : res)
{
cout << '(' << a << ',' << b << ')' << endl;
}
return 0;
}