kuangbin带你飞——专题一 简单搜索(11)
题目来源:POJ 3984 迷宫问题
题解
典型的迷宫问题用BFS(广度优先搜索)
,难点在于将路线再输出出来。
从结果回溯回去输出路线,可以使用递归DFS
。
AC代码
#include <iostream>
#include <queue>
#include <cstring>
using namespace std;
// 点坐标结构体,pre 表示上一个点在 p 数组中位置,now 表示自己在 p 数组中位置
struct point
{
int pre, x, y, now;
};
// 地图数组 mp,标记数组 vis,方向数组 dx 和 dy
int mp[10][10];
int vis[10][10];
int dx[] = {-1, 1, 0, 0},
dy[] = {0, 0, -1, 1};
// 点坐标数组
point p[100];
int cnt = 1;
// 广度优先搜索,返回值为终点坐标结构体
point bfs()
{
// 初始化数组
memset(vis, 0, sizeof(vis));
p[0].pre = -1, p[0].now = 0, p[0].x = 0, p[0].y = 0;
vis[0][0] = 1;
// 起点入队
queue<point> q;
q.push(p[0]);
while (!q.empty())
{
point tmp = q.front();
q.pop();
// 到达终点时,返回终点坐标结构体
if (tmp.x == 4 && tmp.y == 4)
return tmp;
// 遍历四个方向
for (int i = 0; i < 4; ++i)
{
if (!vis[tmp.x + dx[i]][tmp.y + dy[i]] && !mp[tmp.x + dx[i]][tmp.y + dy[i]])
{
vis[tmp.x + dx[i]][tmp.y + dy[i]] = 1;
p[cnt].pre = tmp.now;
p[cnt].x = tmp.x + dx[i];
p[cnt].y = tmp.y + dy[i];
p[cnt].now = cnt;
q.push(p[cnt++]);
}
}
}
}
// dfs 回溯输出结果
void dfs(point ans)
{
if (ans.x == 0 && ans.y == 0)
{
printf("(%d, %d)\n", ans.x, ans.y);
return;
}
dfs(p[ans.pre]);
printf("(%d, %d)\n", ans.x, ans.y);
}
int main()
{
for (int i = 0; i < 5; ++i)
for (int j = 0; j < 5; ++j)
scanf("%d", &mp[i][j]);
dfs(bfs());
return 0;
}