题目来自胡凡的算法笔记
一、题目描述
这个题目是算法笔记中,用来讲解BFS(广度优先搜索)所使用的案例,作者胡凡给出了BFS的解题代码和思路;
二、BFS、DFS解题方法和代码
(1)BFS思路代码
思路:
其实很好理解,BFS就是按照每一层每一层的结构来往下进行搜索,比如第矩阵的第一个元素满足搜索条件,那么就对它周围的四个点挨个进行判断(如果满足条件就将他们入队),这样按照入队顺序进行继续进行搜索,直至队列中没有元素,就结束此次广度优先搜索。
可以结合下边的图来理解,第一次进入迷宫,A就是满足条件的点,然后它的周围有BC,且满足搜索条件,就将他们入队,入队后A就搜索完了,将A出队就可以了,这样再继续进行队列中后边的元素的搜索。
BFS代码:
#include<cstdio>
#include<queue>
using namespace std;
#define maxn 100
struct node{ //结点的x、y坐标
int x;
int y;
}Node;
bool inq[maxn][maxn] = {false}; //用来作为判断这个结点是否入过队列
int matrix[maxn][maxn]; //用来存放01矩阵
int X[4] = {0,0,-1,1};
int Y[4] = {1,-1,0,0};//两个矩阵竖着看 按照上下 左右的顺序来判断是否入队
int n,m;//n代表行,m代表列
bool judge(int x,int y){//用来判断是否已经入队和访问是否有效(判断是否过界)
if(x < 0 || x >= n || y < 0 || y >= m) return false; //过界就返回false
else if(matrix[x][y] == 0 || inq[x][y] == true) return false; //如果这个点是0 或者这个点已经入队了 就返回false
else return true;
}
void BFS(int x,int y){//进行广度优先搜索,将每个 1 的结点的 inq 都设置为 true
queue<node> Q;//现在这一层的队列
Node.x = x,Node.y = y; //现在这一结点的坐标
Q.push(Node);
inq[x][y] = true; //将(x,y)设置为已入过队
while(!Q.empty()){
node top = Q.front(); //取出队首元素
Q.pop();
for(int i = 0;i < 4;i++){
int newX = top.x + X[i];
int newY = top.y + Y[i];
if(judge(newX,newY)){ //如果这一点满足入队条件
Node.x = newX;
Node.y = newY;
Q.push(Node); //将这个结点入队
inq[newX][newY] = true; //设置入队
}
}
}
}
int main(){
scanf("%d%d",&n,&m);
for(int i=0;i<n;i++){
for(int j=0;j<m;j++){
scanf("%d",&matrix[i][j]); //按照行列输入数据
}
}
int ans;//存放块数
for(int i=0;i<n;i++){
for(int j=0;j<m;j++){//对对每个没有入队的点进行BFS
if(matrix[i][j] == 1 && inq[i][j] == false){
ans++;
BFS(i,j);
}
}
}
printf("%d",ans); //输出结果
return 0;
}
(2)DFS思路、代码
思路
DFS和BFS的搜索判断条件都是一样的,所以我们可以沿用BFS的判断函数,但是搜索过程相对于BFS来说会简单些。
如果矩阵第一个点是满足条件的点,那么我们就依次对它周围的四个(或两个)进行判断(满足搜索条件就对这个点进行DFS,一直到无法继续就返回),这样每次判断都会把这个点周围和周围点的周围是 1 的都搜索出来,每个搜索出来的点就把 inq[][ ]矩阵中的此点对应的坐标置 true,后边就不会进行重复搜索了。
过程可以结合下边的图来理解:
从A出发,A可以到的有B和C,先去B,而B又有岔路,这样重复A的操作,直至遇到"死胡同"就返回,进入下一个岔路。
DFS代码:
#include<cstdio>
#define maxn 100
using namespace std;
int n = 0,m = 0; //n 和 m分别代表矩阵的行和列
int ans=0; //用来存储结果
int matrix[6][7] = {{0,1,1,1,0,0,1},{0,0,1,0,0,0,0},{0,0,0,0,1,0,0},{0,0,0,1,1,1,0},{1,1,1,0,1,0,0},{1,1,1,1,0,0,0}};//待搜索的01矩阵
int X[4] = {0,0,-1,1};
int Y[4] = {1,-1,0,0}; //X 和 Y两个矩阵竖着看, 对这个结点进行前后左右的搜索
bool inq[maxn][maxn] = {false}; //用来判断这个当前搜索的结点是否已经被搜索过
bool judge(int x,int y){//用来判断是不是可以搜索
if(x < 0 || x > n || y < 0 || y > m) return false; //如果已经越界就返回false
else if(inq[x][y] == true || matrix[x][y] == 0) return false; //如果已经搜索过或者不满足搜索条件,就返回false
else return true;
}
void DFS(int nowX,int nowY){
inq[nowX][nowY] = true; //说明这一块已经搜索过了
for(int i = 0;i < 4;i++){
int x = nowX + X[i];
int y = nowY + Y[i];//判断周围的点
if(judge(x,y)){
DFS(x,y); //后边不需要加return,加了以后会提早结束这个循环,会导致错误
}
}
}
int main(){
scanf("%d%d",&n,&m); //输入行列数
//因为前边把01矩阵先定义了 所以不需要输入了 可以根绝需要更改
for(int i=0;i<n;i++){
for(int j=0;j<m;j++){//先输入矩阵
if(judge(i,j)){ //如果满足搜索条件
ans++;
DFS(i,j); //向下搜索所有相邻的点
}
}
}
printf("%d",ans);
return 0;
}