在给定的网格中,每个单元格可以有以下三个值之一:
值 0 代表空单元格;
值 1 代表新鲜橘子;
值 2 代表腐烂的橘子。
每分钟,任何与腐烂的橘子(在 4 个正方向上)相邻的新鲜橘子都会腐烂。
返回直到单元格中没有新鲜橘子为止所必须经过的最小分钟数。如果不可能,返回 -1。
- 示例
输入:[[2,1,1],[1,1,0],[0,1,1]]
输出:4
思路
假设图中只有一个腐烂的橘子,它每分钟向外拓展,腐烂上下左右相邻的新鲜橘子,那么下一分钟,就是这些被腐烂的橘子再向外拓展腐烂相邻的新鲜橘子,这与广度优先搜索的过程均一一对应,上下左右相邻的新鲜橘子就是该腐烂橘子尝试访问的同一层的节点,路径长度就是新鲜橘子被腐烂的时间。因此我们使用BFS
求解,过程如下。
- 找到所有腐烂的橘子,加入队列,作为第0层的节点
- 进行遍历,将上下左右四个方向的新鲜橘子进行腐烂,注意边界条件。
- 由于可能存在无法被污染的橘子,我们需要记录新鲜橘子的数量。在 BFS 中,每遍历到一个橘子(污染了一个橘子),就将新鲜橘子的数量减一。如果 BFS 结束后这个数量仍未减为零,说明存在无法被污染的橘子。
- 代码
public int orangesRotting(int[][] grid) {
int m = grid.length;
int n = grid[0].length;
int[][] dirs = {{-1,0},{1,0},{0,-1},{0,1}};
Queue<int[]> queue = new LinkedList<>();
// 记录新鲜橘子的数量
int count = 0;
for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++) {
if (grid[i][j]==1)
count++;
//橘子是腐烂的,加入队列
else if (grid[i][j]==2)
queue.offer(new int[]{i,j});
}
}
//分钟数,也可以说是腐烂的轮数
int round = 0;
// 不加count>0这个条件,结果会多1.
while (count>0 && !queue.isEmpty()){
round++;
int size = queue.size();
//遍历每一层
for (int i = 0; i < size; i++) {
int[] poll = queue.poll();
//遍历每个节点的上下左右四个方向
for (int j = 0; j < 4; j++) {
int x = poll[0]+dirs[j][0];
int y = poll[1]+dirs[j][1];
//注意边界条件
if (x>=0 && x<m && y>=0 && y<n && grid[x][y]==1){
grid[x][y] = 2;
count--;
queue.offer(new int[]{x,y});
}
}
}
}
//说明还有新鲜橘子
if (count!=0)
return -1;
return round;
}