BFS 广度优先遍历
题目
给定一个由 0 和 1 组成的矩阵,找出每个元素到最近的 0 的距离。两个相邻元素间的距离为 1 。
示例 1:
输入:
0 0 0
0 1 0
1 1 1
输出:
0 0 0
0 1 0
1 2 1
注意:
给定矩阵的元素个数不超过 10000。
给定矩阵中至少有一个元素是 0。
矩阵中的元素只在四个方向上相邻: 上、下、左、右。
看到题是懵的,自己思考了一会发现只能写出来一部分,就看了评论区,知道要用BFS,但是自从大二学完数据结构就没有在看过了,完全不会,就去查资料,明白以后就做出来题;
广度优先遍历(BFS)
visited数组:0表示未访问,1表示以访问。
步骤
1、遍历从节点V0开始,访问节点V0,V0入队,visited[0] = 1;
2、访问V0邻节点V2,V2入队,visited[2]=1;
3、访问V0邻节点V1,V1入队,visited[1]=1;
4、访问V0邻节点V3,V3入队,visited[3]=1;到此为止V0的邻节点访问完毕,V0出队;
5、访问V2的邻节点V4,V4入队,visited[4] = 1;到此为止V2的邻节点访问完毕,V2出队;
6、访问V1的邻节点V4,由于visited[4]=1,已经被访问,不能访问;
7、访问V1的邻节点V5,visited[5]=1,到此为止V1的邻节点访问完毕,V1出队;
8、访问V3的邻节点V5,由于visited[5]=1,已经被访问,不能访问;到此为止V3的邻节点访问完毕,V3出队;
9、访问V4的邻节点V6,visited[6]=1,V6入队;到此为止V4的邻节点访问完毕,V4出队;
10、访问V5的邻节点V6,由于visited[6]=1,已经被访问,不能访问;到此为止V5的邻节点访问完毕,V5出队;
11、V6无邻节点,V6出队;遍历完毕;
题目分析
0 0 0
0 1 0
1 1 1
- 从V00开始遍历,当其等于0时将其入队,否则将其设置一个极大地数字如:10000;
- 将第一个数值出队,进行四个方向的遍历,若相邻节点存在,且相邻节点的值小于等于当前值+1则返回,(用于判断是否已经入队,也可以设置一个boolean数组用于判断是否已经入队)否则赋值为1,入队
if (x < 0 || x >= row || y < 0 || y >= col || matrix[x][y] <= matrix[node[0]][node[1]] + 1) {
continue;
}
举例:
在这里插入图片描述
- V00,V01,V02,V10,V12入队
- V00进行遍历,向下遍历相邻节点为V10=0,V10<=V00+1,continue;
- V00进行遍历,向左遍历相邻节点为V01=0,V01<=V00+1,continue;V00出队
- V01进行遍历,向下遍历相邻节点为V11=10000,V10<=V01+1不满足,则V11=1;V11入队;
- V01进行遍历,向左遍历相邻节点为V02=0,V02<=V01+1,continue;V00出队;
依次类推得到最后的值;
public static int[][] uatpdeMatrix(int[][] matrix) {
/**
BFS思想, 把所有1都置为最大值, 把所有为0的位置加入队列中, 每次从队列中poll
一个节点, 更新其四周的节点, 如果被更新的节点距离变小了就将其入队列准备更新其邻接点
**/
int row = matrix.length;
if (row < 1) {
return matrix;
}
int col = matrix[0].length;
Queue<int[]> q = new LinkedList<>();
for (int i = 0; i < row; ++i) {
for (int j = 0; j < col; ++j) {
if (matrix[i][j] == 1) {
matrix[i][j] = Integer.MAX_VALUE;
} else {
q.offer(new int[]{i, j});
}
}
}
int[][] dirs = {{0, -1}, {0, 1}, {-1, 0}, {1, 0}};//四个方向
while (!q.isEmpty()) {
int[] node = q.poll();
for (int[] dir : dirs) {
int x = node[0] + dir[0];
int y = node[1] + dir[1];
//如果Matrix[x][y]不存在,当前值matrix[x][y]比matrix[node[0]][node[1]]+1小(判断0/1)一个节点只入队列一次
if (x < 0 || x >= row || y < 0 || y >= col || matrix[x][y] <= matrix[node[0]][node[1]] + 1) {
continue;
}
//当前值为1
matrix[x][y] = matrix[node[0]][node[1]] + 1;
q.offer(new int[]{x, y}); //入队
}
}
return matrix;
}