描述
给定一个 n×m n×m 的网格,其中每个单元格中可能有三种值中的一个 0 , 1 , 2。其中 0 表示这个格子为空、1 表示这个格子有一个完好的苹果,2 表示这个格子有一个腐烂的苹果。腐烂的苹果每分钟会向上下左右四个方向的苹果传播一次病菌,并导致相邻的苹果腐烂。请问经过多少分钟,网格中不存在完好的苹果。如果有苹果永远不会腐烂则返回 -1
数据范围: 1≤n,m≤1000 1≤n,m≤1000 ,网格中的值满足 0≤val≤2 0≤val≤2
示例1
输入:
[[2,1,1],[1,0,1],[1,1,1]]
复制返回值:
4
复制
示例2
输入:
[[2,1,0],[1,0,1],[0,0,0]]
复制返回值:
-1
思路
第一步我们先要明白,每个腐烂的苹果都会同时开始向周围腐烂
第二步,我们把所有坏的苹果放进栈,同时出栈,开始腐烂 ,也就是广度优先遍历
第三步,我们开始记录被腐烂的苹果,即可,切记我们要把被腐烂的苹果也就是新腐烂的苹果放进栈,这里你可能想为啥不把旧腐烂苹果重新放进栈去呢?那我反问你,旧腐烂的苹果周边都被腐烂了,还用重新扩散取腐烂吗
代码
import java.util.*;
public class Solution {
int []dx = {0, 0, 1, -1};
int [] dy = {1, -1, 0, 0};
boolean [][]vis;//标记腐烂的苹果
public int rotApple (ArrayList<ArrayList<Integer>> list) {
int m = list.size(), n = list.get(0).size();
vis = new boolean[m][n];
int count = 0;
//把腐烂的苹果放进栈
Queue<int []>q = new LinkedList<>();
for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++) {
if (list.get(i).get(j) == 2)q.add(new int[] {i, j});
}
}
//因为每个腐烂的苹果都是同时开始传染,所以同时出栈广度优先遍历
while (!q.isEmpty()) {
int sz = q.size();
while (sz-- != 0) {
int []t=q.poll();
int a = t[0], b = t[1];
for (int k = 0; k < 4; k++) {
int x = a + dx[k], y = b + dy[k];
if (x >= 0 && x < m && y >= 0 && y < n && !vis[x][y] &&
list.get(x).get(y) == 1) {
//找到一个好苹果开始腐烂
vis[x][y] = true;
q.add(new int [] {x, y});
}
}
}
count++;
}
//判断是否有永远不会腐烂的苹果
for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++) {
if (vis[i][j] == false&&list.get(i).get(j)==1)return -1;
}
}
return count-1;
}
}