leetcode 1020. Number of Enclaves(内飞地数量)

leetcode 1020. Number of Enclaves(内飞地数量)_第1张图片
飞地的概念可自行baidu.

数组grid中0代表海,1代表陆地。
在陆地上可以上下左右4个方向移动,也可以走出数组边界。
返回grid中经过移动不能走出数组边界的元素个数。

思路:

和leetcode 1254题很像,区别是1254题只是求多少个连通区域,而本题求的是多少个cell.

不能像1254题那样只访问grid中间部分的元素,因为中间部分可能是飞地,也可能不是飞地,在求完一个连通区域之前,并不确定哪个cell是飞地。只能得到连通区域的个数,并不能得到cell的个数。

所以转变思路,从能出去的边界部分做dfs. 边界部分能访问到的肯定不是飞地。
最后没有访问过的,且元素为1的就是飞地。

class Solution {
    int rows = 0;
    int cols = 0;
  
    public int numEnclaves(int[][] grid) {
        rows = grid.length;
        cols = grid[0].length;
        
        int res = 0;

        if(rows < 3 || cols < 3) return 0;

        for(int i = 0; i < rows; i++) {
            for(int j = 0; j < cols; j++) {
                if((i == 0 || i == rows-1 || j == 0 || j == cols-1) && grid[i][j] == 1) {
                    dfs(grid, i, j);
                } 
            }                  
        }
        

        for(int i = 0; i < rows; i++) {
            for(int j = 0; j < cols; j++) {
                if(grid[i][j] == 1) res ++;
            }
        }
        return res;
    }

    void dfs(int[][] grid, int r, int c) {
        if(r < 0 || r > rows-1 || c < 0 || c > cols-1) return;

        if(grid[r][c] != 1) return;
      
        grid[r][c] = -1;

        dfs(grid, r, c-1);
        dfs(grid, r, c+1);
        dfs(grid, r-1, c);
        dfs(grid, r+1, c);
    }
}

你可能感兴趣的:(leetcode,深度优先,算法)