运用广度优先搜索(BFS)
class Solution { public int orangesRotting(int[][] grid) { int count=0;//记录新鲜橘子 int N = grid.length;//记录数组的行 int M = grid[0].length;//记录数组的列 Queue<int[]> queue = new LinkedList<>(); for(int i=0;i<N;i++){ for(int j=0;j<M;j++){ if(grid[i][j]==1){ count++; }else if(grid[i][j]==2){ queue.add(new int[]{i,j}); } } } int time = 0;//记时间 while(!queue.isEmpty() && count>0){ time++; int n = queue.size(); for(int i=0;i<n;i++){ int[] arr = queue.poll(); int a = arr[0];//代表烂橘子的左右 int b = arr[1];//代表烂橘子的上下 if(a-1>=0 && grid[a-1][b]==1){ count--; grid[a-1][b] = 2; queue.add(new int[]{a-1,b}); } if(a+1<N && grid[a+1][b]==1){ count--; grid[a+1][b] = 2; queue.add(new int[]{a+1,b}); } if(b-1>=0 && grid[a][b-1]==1){ count--; grid[a][b-1] = 2; queue.add(new int[]{a,b-1}); } if(b+1<M && grid[a][b+1]==1){ count--; grid[a][b+1] = 2; queue.add(new int[]{a,b+1}); } } } if(count>0){ return -1; }else { return time; } } }
java
运行