《算法笔记》读书记录DAY_29

CHAPTER_8  提高篇(2)——搜索专题

8.2 广度优先搜索(BFS)

广度优先搜索也是用来遍历或搜索树和图数据结构的算法，它和深度优先搜索的思想截然不同。广度优先搜索和二叉树的层序遍历较为相似，需要借助于队列实现；深度优先搜索和二叉树的前序遍历较为相似，需要借助于栈实现。

BFS以广度为第一关键词，它总是访问当前邻接的所有结点，再按照这些邻接结点被访问的顺序，再依次访问它们能到达的邻接节点，直到所有结点被访问结束。

我们来看一个BFS例子，从v1开始遍历下面这个图。首先访问v1，v1访问结束后即将访问它的3个邻接结点(v2、v3、v4)，我们在访问v1时将这三个节点记录下来。我们先访问v2，同样的记录v2的邻接节点v6，待前面所有的记录节点访问完成时访问v6。v2访问结束，访问v3，记录v3的所有邻接节点(v5)，v5的访问顺序在v6后面。v3访问结束，访问v4，记录v4的邻接节点v7。此时v1的三个邻接节点访问完毕，我们开始第三层的访问了，也就是v2、v3、v4的邻接节点访问。按照记录顺序，我们访问v6，此刻v6已经没有未记录的邻接节点了。然后访问v5，也没有邻接节点，之后访问v7，v7同样没未记录的邻接节点。至此，所有节点访问结束。

可以看到，整个算法的过程很像一个队列。记录节点的过程就是将所有待处理的节点入队，然后按照先进先出的顺序处理节点，再将节点的邻接节点入队，直到所有节点入队并处理完。

因此BFS可以用队列实现，我们给出基本模板如下:

void BFS() {
	queue<int> q;
	q.push(s);
	while(!q.empty()) {
		取出队首元素； 
		访问队首元素； 
		将队首元素出队；
		将top的下一层结点中未入队的入队，并设置未已入队 
	}
} 

接下来看一道例题，来熟悉BFS的队列实现。

题目：

给出一个m*n型矩阵，矩阵中的元素为0或1。称位置(x,y)与其上下左右(x,y+1)、(x+1,y)、(x-1,y)、(x,y-1)四个点是相邻的。如果矩阵中有若干个1是相邻的(不必两两相邻)，称这些"1"构成了一个块。求给定矩阵中块的个数。

输入样例：

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 

输出样例： 

4 

思路： 

枚举每一个位置的元素，如果为0，则跳过该元素；如果为1，则使用BFS查询与该位置相邻的四个位置(注意不要出边界)，判断它们是否为1，如果某个相邻元素为1，则继续查询它的相邻元素，如此往复直到所有1被访问完毕。同时要设置一个数组，标志某个位置的1是否被访问。

参考代码：

#include <iostream>
#include <queue>
#include <utility>
using namespace std;

const int maxn=100;
int matrix[maxn][maxn],m,n;                        //matrix存储m*n矩阵
bool flag[maxn][maxn]={0};                         //记录每个位置是否入队 

void BFS(int x,int y) {
	queue<pair<int,int> > q;
	q.push(make_pair(x,y));                        //将点(x,y)入队 
	flag[x][y]=1;                                  //将(x,y)标记为已访问 
	while(!q.empty()) {
		int xPos=q.front().first;
		int yPos=q.front().second;                 //访问队首元素，获得点(xPos,yPos) 
		q.pop();                                   //队首出队 
		if(xPos-1>=0&&flag[xPos-1][yPos]==0&&matrix[xPos-1][yPos]==1) {   //左邻接点(Pos-1,yPos)要满足三个条件才可入队 
			 q.push(make_pair(xPos-1,yPos));
			 flag[xPos-1][yPos]=1;
		}
		if(xPos+1<=n-1&&flag[xPos+1][yPos]==0&&matrix[xPos+1][yPos]==1) {   //右邻接点(Pos+1,yPos)要满足三个条件才可入队 
			 q.push(make_pair(xPos+1,yPos));
			 flag[xPos+1][yPos]=1;
		}
		if(yPos-1>=0&&flag[xPos][yPos-1]==0&&matrix[xPos][yPos-1]==1) {   //上邻接点(Pos,yPos-1)要满足三个条件才可入队 
			 q.push(make_pair(xPos,yPos-1));
			 flag[xPos][yPos-1]=1;
		}
		if(yPos+1<=n&&flag[xPos][yPos+1]==0&&matrix[xPos][yPos+1]==1) {   //下邻接点(Pos,yPos+1)要满足三个条件才可入队 
			 q.push(make_pair(xPos,yPos+1));
			 flag[xPos][yPos+1]=1;
		}
	}
}

int main() {
	int ans=0;
	cin>>m>>n;
	for(int i=0;i<m;i++) {          //输入矩阵 
		for(int j=0;j<n;j++)
			cin>>matrix[i][j];
	}
	for(int i=0;i<m;i++)
		for(int j=0;j<n;j++)
			if(matrix[i][j]==1&&flag[i][j]==0) {    //如果元素为1且未被访问 
				BFS(i,j);
				ans++;
			}
	cout<<endl<<ans<<endl;
	return 0;
}