OJ---腐烂的橘子

运用广度优先搜索(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;
        }
    }
}

你可能感兴趣的:(OJ,java,算法)