寒假算法学习（BFS的一些题解）

前面我们学习了BFS的简单概念，这里讲解一些其他的题目加深理解。

题目1：

P1443 马的遍历 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

题目描述

有一个 n×m 的棋盘，在某个点 (x,y) 上有一个马，要求你计算出马到达棋盘上任意一个点最少要走几步。

输入格式

输入只有一行四个整数，分别为 n,m,x,y。

输出格式

一个 n×m 的矩阵，代表马到达某个点最少要走几步（不能到达则输出 −1）。

输入输出样例

输入 

3 3 1 1

输出

0    3    2    
3    -1   1    
2    1    4    

思路：

题中所说的马是象棋中的马，走日（从样例看出）；

不能到达的点输出-1，则全部初始化为-1；

用一个数组记录马到达某一点行动几次，即搜索多少次搜到点；

最后打印该数组即可

代码：

#include<bits/stdc++.h>  //万能头文件
using namespace std;
#define M 410
#define x first			//宏定义 first和 second
#define y second
typedef pair<int ,int> PII;
queue <PII> q;
int n,m,x,y;
int dist[M][M];			//记录马到达某一点行动几次，即搜索多少次搜到该点

int dx[8]={-2,-1,1,2,2,1,-1,-2};  //马移动的八个方位
int dy[8]={1,2,2,1,-1,-2,-2,-1};

void bfs(int x1,int y1){
	memset(dist,-1,sizeof(dist));  //初始化为-1
	q.push({x1,y1});  //入队
	dist[x1][y1]=0;   //起点标记为 0

	while(q.size()){  //对列不空
		auto t=q.front();    //取出对头
		q.pop();			//弹出对头
		for(int i=0;i<8;i++){
			int a=t.x+dx[i];
			int b=t.y+dy[i];

			if(a<1||a>n||b<1||b>m)continue;  //控制在边界之内
			if(dist[a][b]>=0)continue;		//走过该点

			q.push({a,b});					//入队
			dist[a][b]=dist[t.x][t.y]+1;	//行动次数 +1
		}
	}
	return ;
}
int main(){
	scanf("%d%d%d%d",&n,&m,&x,&y);
	bfs(x,y);
	for(int i=1;i<=n;i++){
		for(int j=1;j<=m;j++){
			printf("%d ",dist[i][j]);
		}
		printf("\n");
	}
	return 0;
}

代码（数组实现对列）：

代码的注释和上面差不多，可参考上述代码

#include<iostream>		//也可用万能头文件
#include<algorithm>
#include<cstring>
#define M 410
#define x first
#define y second
using namespace std;

typedef pair<int ,int> PII;
int n,m;
int map[M][M];
PII q[M*M];		//定义数组，存对列

int dx[8]={-2,-1,1,2,2,1,-1,-2};
int dy[8]={1,2,2,1,-1,-2,-2,-1};

void bfs(int x1,int y1){
	for(int i=1;i<=n;i++){
		for(int j=1;j<=m;j++){
			map[i][j]=-1;
		}
	}
	q[0]={x1,y1};
	map[x1][y1]=0;
	int hh=0,tt=0;		//hh指向对头，tt指向对尾
	while(hh<=tt){		//对列不空
		auto t=q[hh];  //取出队头，相当于q.front()
		hh++;    //弹出队头,相当于q.pop()
		for(int i=0;i<8;i++){
			int a=t.x+dx[i],b=t.y+dy[i];
			if(a<1||a>n||b<1||b>m)continue;
			if(map[a][b]>=0)continue;
			map[a][b]=map[t.x][t.y]+1;
			q[++tt]={a,b};
		}
	}
}
int main(){
	int x1,y1;
	scanf("%d%d%d%d",&n,&m,&x1,&y1);
	bfs(x1,y1);
	for(int i=1;i<=n;i++){
		for(int j=1;j<=m;j++){
			printf("%5d ",map[i][j]);
		}
		printf("\n");
	}
	return 0;
}

题目2：

P1332 血色先锋队 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

题目背景

巫妖王的天灾军团终于卷土重来，血色十字军组织了一支先锋军前往诺森德大陆对抗天灾军团，以及一切沾有亡灵气息的生物。孤立于联盟和部落的血色先锋军很快就遭到了天灾军团的重重包围，现在他们将主力只好聚集了起来，以抵抗天灾军团的围剿。可怕的是，他们之中有人感染上了亡灵瘟疫，如果不设法阻止瘟疫的扩散，很快就会遭到灭顶之灾。大领主阿比迪斯已经开始调查瘟疫的源头。原来是血色先锋军的内部出现了叛徒，这个叛徒已经投靠了天灾军团，想要将整个血色先锋军全部转化为天灾军团！无需惊讶，你就是那个叛徒。在你的行踪败露之前，要尽快完成巫妖王交给你的任务。

题目描述

军团是一个 n 行 m 列的矩阵，每个单元是一个血色先锋军的成员。感染瘟疫的人，每过一个小时，就会向四周扩散瘟疫，直到所有人全部感染上瘟疫。你已经掌握了感染源的位置，任务是算出血色先锋军的领主们感染瘟疫的时间，并且将它报告给巫妖王，以便对血色先锋军进行一轮有针对性的围剿。

输入格式

第 1 行：四个整数 n，m，a，b，表示军团矩阵有 n 行 m 列。有 a 个感染源，b 为血色敢死队中领主的数量。

接下来 a 行：每行有两个整数 x，y，表示感染源在第 x 行第 y 列。

接下来 b 行：每行有两个整数 x，y，表示领主的位置在第 x 行第 y 列。

输出格式

第 1 至 b 行：每行一个整数，表示这个领主感染瘟疫的时间，输出顺序与输入顺序一致。如果某个人的位置在感染源，那么他感染瘟疫的时间为 0。

输入输出样例

输入

5 4 2 3
1 1
5 4
3 3
5 3
2 4

输出

3
1
3

说明/提示

输入输出样例

如下图，标记出了所有人感染瘟疫的时间以及感染源和领主的位置。

代码： 

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;

#define x first
#define y second
#define M 510

typedef pair<int,int> PII;
PII q[M*M];
int hh=0,tt=-1;		//hh指向对头，tt指向队尾
int n,m,a,b;
int map[M][M];
int dist[M][M];		//记录搜索次数

int dx[4]={-1,0,1,0};	//四个方向
int dy[4]={0,1,0,-1};

void bfs(){
	while(hh<=tt){
		auto t =q[++hh];		//取出对头，再弹出对头

		for(int i=0;i<4;i++){
			int a=t.x+dx[i];
			int b=t.y+dy[i];

			if(a<1||a>n||b<1||b>m)continue;		//限定边界
			if(dist[a][b]>=0)continue;

			dist[a][b]=dist[t.x][t.y]+1;
			q[++tt]={a,b};
		}
	}
}

int main(){
	scanf("%d%d%d%d",&n,&m,&a,&b);
	memset(dist,-1,sizeof(dist));
	while(a--){
		int x1,y1;
		scanf("%d%d",&x1,&y1);
		q[++tt]={x1,y1};		//输入一个感染源就入队
		dist[x1][y1]=0;			//入对的点，记录为 0
	}
	bfs();
	while(b--){
		int x1,y1;
		scanf("%d%d",&x1,&y1);	//输入一个领主坐标，就输出答案
		printf("%d\n",dist[x1][y1]);
	}
	return 0;
}

题目3：

173. 矩阵距离 - AcWing题库

给定一个 N行 M列的 01 矩阵 A，A[i][j]与 A[k][l]之间的曼哈顿距离定义为：

dist(A[i][j],A[k][l])=|i−k|+|j−l|

输出一个 N 行 M列的整数矩阵 B，其中：

B[i][j]=min1≤x≤N,1≤y≤M,A[x][y]=1dist(A[i][j],A[x][y])

输入格式

第一行两个整数 N,M。

接下来一个 N 行 M列的 01 矩阵，数字之间没有空格。

输出格式

一个 N 行 M列的矩阵 B，相邻两个整数之间用一个空格隔开。

数据范围

1≤N,M≤1000

输入样例：

3 4
0001
0011
0110

输出样例：

3 2 1 0
2 1 0 0
1 0 0 1

代码： 

#include<bits/stdc++.h>
using namespace std;
#define M 1010
#define x first
#define y second
typedef pair<int,int> PII;
PII q[M*M];			//用数组实现对列
int n,m;
char g[M][M];
int dist[M][M];
int hh=0,tt=-1;		//hh指向对头，tt指向队尾

int dx[4]={-1,0,1,0};
int dy[4]={0,1,0,-1};	//四个方向

void bfs(){
	while(hh<=tt){			//对列不空
		auto t=q[hh++];		//取出对头，再弹出对头

		for(int i=0;i<4;i++){	//遍历四个方向
			int a=t.x+dx[i];
			int b=t.y+dy[i];

			if(a<1||a>n||b<1||b>m)continue;		//限定在边界之内
			if(dist[a][b]>=0)continue;

			dist[a][b]=dist[t.x][t.y]+1;		//满足条件，搜索次数 +1
			q[++tt]={a,b};
		}
	}
}
int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++){
		scanf("%s",g[i]+1);
	}
	memset(dist,-1,sizeof(dist));		//初始化为 -1
	for(int i=1;i<=n;i++){
		for(int j=1;j<=m;j++){
			if(g[i][j]=='1'){
				q[++tt]={i,j};
				dist[i][j]=0;
			}
		}
	}
	bfs();
	for(int i=1;i<=n;i++){
		for(int j=1;j<=m;j++){
			printf("%d ",dist[i][j]);
		}
		printf("\n");
	}
	return 0;
}