思路:BFS
是典型的BFS题型,只不过这里用来表示腐烂的传染,初次碰见的读者可以这样想:
如果一个人生病了,他会传染给其他跟他接近的人,然后接近他的人感冒了就会传染给其他接近他们的人,依次类推。
这道题就是这样传染病模型,我们实现BFS需要定义队列来实现,初始化的时候首先把0分钟就腐烂的橘子装入队列中,然后再进行BFS遍历比较方便。然后,又定义两个移动数组dy,dx用来判断周围的格子是否符合条件入队。count数组则是记录每个格子橘子腐烂时的时间。因为BFS遍历比较简单,可以直接找模板,这里直接贴代码不详细讲了。
注意:这里作者把0分钟腐烂的橘子在count数组定义中初始化了1,因为一开始创建count数组的时候初始值都是0,这样用于分辨。到最后遍历count最大值之后,再在结果上-1就行了。
如果本来就没有腐烂的橘子,直接返回0.
class Solution {
class Node{
int x;
int y;
public Node(int x,int y){
this.x=x;
this.y=y;
}
}
int [][]count;
int []dx={-1,1,0,0};
int []dy={0,0,-1,1};
Deque<Node>q=new LinkedList<>();
public void bfs(int [][]grid){
while(!q.isEmpty()){
Node n=q.getFirst();
q.removeFirst();
for(int i=0;i<4;i++){
int a=n.x+dx[i];
int b=n.y+dy[i];
if(a<0||a>=grid.length||b<0||b>=grid[0].length)continue;
if(grid[a][b]==0)continue;
if(count[a][b]>0)continue;
count[a][b]=count[n.x][n.y]+1;
Node node=new Node(a,b);
q.addLast(node);
}
}
}
public int orangesRotting(int[][] grid) {
count=new int[grid.length][grid[0].length];
for(int i=0;i<grid.length;i++){
for(int j=0;j<grid[0].length;j++){
if(grid[i][j]==2){
count[i][j]=1;
Node node=new Node(i,j);
q.addLast(node);
}
}
}
bfs(grid);
int res=0;
for(int i=0;i<count.length;i++){
for(int j=0;j<count[0].length;j++){
if(grid[i][j]==1&&count[i][j]==0){
return -1;
}
res=Math.max(res,count[i][j]);
}
}
if(res==0)
return 0;
else
return res-1;
}
}