Leetcode No.200 岛屿数量（DFS）

原创

姜兴琪 2022-01-19 13:48:27 ©著作权

文章标签 leetcode DFS 深度优先搜索二维 i++ 文章分类 代码人生

©著作权归作者所有：来自51CTO博客作者姜兴琪的原创作品，请联系作者获取转载授权，否则将追究法律责任

一、题目描述

给你一个由 '1'（陆地）和 '0'（水）组成的的二维网格，请你计算网格中岛屿的数量。

岛屿总是被水包围，并且每座岛屿只能由水平方向和/或竖直方向上相邻的陆地连接形成。

此外，你可以假设该网格的四条边均被水包围。

示例 1：

输入：grid = [

["1","1","1","1","0"],

["1","1","0","1","0"],

["1","1","0","0","0"],

["0","0","0","0","0"]

]

输出：1

示例 2：

输入：grid = [

["1","1","0","0","0"],

["0","0","1","0","0"],

["0","0","0","1","1"]

]

输出：3

提示：

m == grid.length

n == grid[i].length

1 <= m, n <= 300

grid[i][j] 的值为 '0' 或 '1'

二、解题思路

我们可以将二维网格看成一个无向图，竖直或水平相邻的 1之间有边相连。

为了求出岛屿的数量，我们可以扫描整个二维网格。如果一个位置为 1，则以其为起始节点开始进行深度优先搜索。我们定义辅助数据visited标记岛屿是否被访问过，在深度优先搜索的过程中，每个搜索过的岛屿都会被标记为true。

最终岛屿的数量就是我们进行深度优先搜索的次数。

三、代码

public class Solution {
    int m,n;
    int cnt=0;
    boolean[][] visited;
    public int numIslands(char[][] grid) {
        n=grid.length;
        m=grid[0].length;
        visited=new boolean[n][m];
        if(n==0){
            return 0;
        }
        for(int i=0;i<n;i++){
            for(int j=0;j<m;j++){
                if(visited[i][j]==false){
                    visited[i][j]=true;
                    if(grid[i][j]=='1'){
                        dfs(grid,i,j);
                        cnt++;
                    }
                }
            }
        }
        return cnt;
    }
    void dfs(char[][] grid,int i,int j){
        if(i+1<n&&visited[i+1][j]==false){
            visited[i+1][j]=true;
            if(grid[i+1][j]=='1'){
                dfs(grid,i+1,j);
            }
        }
        if(i-1>=0&&visited[i-1][j]==false){
            visited[i-1][j]=true;
            if(grid[i-1][j]=='1'){
                dfs(grid,i-1,j);
            }
        }
        if(j+1<m&&visited[i][j+1]==false){
            visited[i][j+1]=true;
            if(grid[i][j+1]=='1'){
                dfs(grid,i,j+1);
            }
        }
        if(j-1>=0&&visited[i][j-1]==false){
            visited[i][j-1]=true;
            if(grid[i][j-1]=='1'){
                dfs(grid,i,j-1);
            }
        }

    }

    public static void main(String[] args) {
        char[][] grid = {
                {'1','1','1','1','0'},
                {'1','1','0','1','0'},
                {'1','1','0','0','0'},
                {'0','0','0','0','0'}
        };
        //char[][] grid={{'1','1','1'},{'0','1','0'},{'1','1','1'}};
        Solution solution=new Solution();
        System.out.println(solution.numIslands(grid));
    }
}