每日一题---腐烂的苹果(广度优先搜索)
腐烂的苹果
给定一个 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
经典广度优先遍历题:
这里介绍使用到的元素:
boolean vis[][] 用于记录好的苹果是否被感染
偏移数组
队列,因为需要同时感染,用于同时向四周感染并记录接收感染后的苹果(用于后续感染)
int time 用于记录感染消耗的时间
题解:
1.统计腐烂苹果进入队列
2.每分钟腐烂苹果都会向四周扩散,将队列中的腐烂苹果弹出,并向四周扩散,使用vis记录被感染的苹果,并把被传染的苹果再次加入队列。
3.如果队列中还有元素就继续执行,并使时间加1
4.结合vis[]统计是否还有好的苹果。有的话就输出-1,否则输出time
代码:
import java.util.*;public class Solution {int m, n;int[] dx = {0, 0, 1, -1};int[] dy = {1, -1, 0, 0};boolean[][] vis;public int rotApple (ArrayList<ArrayList<Integer>> grid) {// write code herem = grid.size();n = grid.get(0).size();vis = new boolean[m][n];//标记苹果是否被感染Queue<int[]> q = new LinkedList<int[]>();for(int i = 0; i < m; i++) {for(int j = 0; j < n; j++) {if(grid.get(i).get(j) == 2) {q.add(new int[] {i, j});}}}int time = 0;//感染的时间while(!q.isEmpty()) {int len = q.size();while(len-- != 0) {int[] tmp = q.poll();for(int i = 0; i < 4; i++) {int x = tmp[0] + dx[i];int y = tmp[1] + dy[i];if(x >= 0 && x < m && y >= 0 && y < n &&!vis[x][y] && grid.get(x).get(y) == 1) {vis[x][y] = true;//标记为已感染q.add(new int[] {x, y});//用于继续向后感染}}}time++;}for(int i = 0; i < m; i++) {for(int j = 0; j < n; j++) {if(grid.get(i).get(j) == 1 && !vis[i][j]) {return -1;}}}return time-1;//因为最后一个好苹果再向周围感染一定感染不到}
}