宽度优先搜索(BFS)本质就是一层一层地搜下去,对于符合条件的,就将其存储到队列中去,遍历完一个长度后,再对队列中的元素依次出队继续遍历下去,直至找到答案。
例1:
走迷宫
给定一个 n×m 的二维整数数组,用来表示一个迷宫,数组中只包含 0 或 1,其中 0 表示可以走的路,1 表示不可通过的墙壁。
最初,有一个人位于左上角 (1,1) 处,已知该人每次可以向上、下、左、右任意一个方向移动一个位置。
请问,该人从左上角移动至右下角 (n,m) 处,至少需要移动多少次。
数据保证 (1,1) 处和 (n,m) 处的数字为 0,且一定至少存在一条通路。
输入格式
第一行包含两个整数 n 和 m。
接下来 n 行,每行包含 m 个整数(0 或 1),表示完整的二维数组迷宫。
输出格式
输出一个整数,表示从左上角移动至右下角的最少移动次数。
数据范围
1≤n,m≤100
输入样例:
5 5
0 1 0 0 0
0 1 0 1 0
0 0 0 0 0
0 1 1 1 0
0 0 0 1 0
输出样例:
8
思路:从出发点开始,对四个方向遍历,未出界且符合条件的入队,再对队列中元素依次如此操作,最终得到答案
代码:
#include<bits/stdc++.h>
using namespace std;
typedef pair<int, int> PII;
const int N = 110;
int n, m;
int g[N][N], d[N][N];//g用来存储地图 d用来标记是否走过并记录步数
int dx[] = { 0,0,-1,1 }, dy[] = { 1,-1,0,0 };//四个方向
void bfs() {
queue<PII> q;
q.push({ 0,0 });
memset(d, -1, sizeof(d));
d[0][0] = 0;
while (q.size()) {
auto t = q.front();//指向队头
q.pop();依次出队
for (int i = 0; i < 4; i++) {
int x = t.first + dx[i], y = t.second + dy[i];
if (x >= 0 && x < n && y >= 0 && y < m && g[x][y] == 0 && d[x][y] == -1) {
q.push({ x,y });//符合条件就入队
d[x][y] = d[t.first][t.second] + 1;
}
}
}
}
int main() {
cin >> n >> m;
for (int i = 0; i < n; i++) {
for (int j = 0; j < m; j++) {
cin >> g[i][j];
}
}
bfs();
cout << d[n - 1][m - 1];
return 0;
}
例2:
题目描述
有一个仅由数字0与1组成的n×n格迷宫。若你位于一格0上,那么你可以移动到相邻4格中的某一格1上,同样若你位于一格1上,那么你可以移动到相邻4格中的某一格0上。
你的任务是:对于给定的迷宫,询问从某一格开始能移动到多少个格子(包含自身)。
输入格式
第1行为两个正整数n,m。
下面n行,每行n个字符,字符只可能是0或者1,字符之间没有空格。
接下来m行,每行2个用空格分隔的正整数i,j,对应了迷宫中第i行第j列的一个格子,询问从这一格开始能移动到多少格。
输出格式
m行,对于每个询问输出相应答案。
输入输出样例
输入 #1
2 2 01 10 1 1 2 2
输出 #1
4 4
说明/提示
所有格子互相可达。
对于20%的数据,n≤10;
对于40%的数据,n≤50;
对于50%的数据,m≤5;
对于60%的数据,n≤100,m≤100;
对于100%的数据,n≤1000,m≤100000。
思路:跟上一题差不多,不过这题不是障碍了,是要满足跟上一步不同,另外这一题的时间也有要求,如果对每个输入都BFS一次就会发现超时了,所以我们用数组把答案存起来,对于一个连通块内的所有格子,所能到达的终点都是一样的,所以一个连通块内我们只需要查询一次就可以了
代码:
#include<bits/stdc++.h>
using namespace std;
typedef pair<int, int> PII;
int n, m;
int g[1010][1010];
bool d[1010][1010];
int dx[] = { -1,1,0,0 }, dy[] = { 0,0,-1,1 };
int ans[1010][1010];
int cnt;
void bfs(int x, int y) {
queue<PII> q;
queue<PII> z;
q.push({ x,y });
d[x][y] = true;
while (q.size()) {
auto t = q.front();
q.pop();
for (int i = 0; i < 4; i++) {
int tx = t.first + dx[i], ty = t.second + dy[i];
if (tx > 0 && tx <= n && ty > 0 && ty <= n &&
abs(g[tx][ty] - g[t.first][t.second]) && !d[tx][ty]) {
d[tx][ty] = true;
cnt++;
q.push({ tx,ty });
z.push({ tx,ty });
}
}
}
ans[x][y] = cnt;
while (z.size()) {
auto r = z.front();
ans[r.first][r.second] = cnt;
d[r.first][r.second] = false;
z.pop();
}
d[x][y] = false;
}
int main() {
memset(d, false, sizeof(d)); //memset时间复杂度为O(n)
cin >> n >> m;
char c;
for (int i = 1; i <= n; i++) {
for (int j = 1; j <= n; j++) {
cin >> c;
g[i][j] = c - '0';
ans[i][j] = 0;
}
c = getchar();
}
int x, y;
while (m--) {
cin >> x >> y;
cnt = 1;
if (!ans[x][y]) {
bfs(x, y);
cout << cnt << endl;
}
else cout << ans[x][y] << endl;
}
return 0;
}
总结:相比DFS,BFS更快,但是需要牺牲空间来换取时间。