「笔试刷题」:岛屿数量

55 阅读1分钟

一、题目

描述

给一个01矩阵,1代表是陆地,0代表海洋, 如果两个1相邻,那么这两个1属于同一个岛。我们只考虑上下左右为相邻。

岛屿: 相邻陆地可以组成一个岛屿(相邻:上下左右) 判断岛屿个数。

例如:

输入

[

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

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

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

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

[0,0,1,1,1]

]

对应的输出为3

(注:存储的01数据其实是字符'0','1')

示例1

输入:

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

返回值:

3

示例2

输入:

[[0]]

返回值:

0

示例3

输入:

[[1,1],[1,1]]

返回值:

1

备注:

01矩阵范围<=200*200

二、思路解析

本题也是一道递归搜索的题,无论是 DFS 还是 BFS 都可以做出来,我则选择了 DFS 的解法。

首先,遍历整个矩阵,对每个未访问过的陆地(即 grid[i][j] 为 1 且未被访问过),进行 DFS 搜索,将与该陆地相邻的所有陆地标记为已访问,并且递归地进行 DFS 搜索。

在 DFS 搜索过程中,我们使用一个辅助数组 vis 来记录已经访问过的位置,防止重复访问。

最后,统计 DFS 调用的次数,即为岛屿的数量,因为在 dfs 方法中,我们将所有陆地的下标 i、j,在 vis 数组中标记为了 true 。

这样处理后,下标 i 和 j 在主方法 main 中的遍历就无法再次进入,进而使 ret++ 了。

然后我们将 ret 作为结果返回即可。

具体实现请看下面代码👇

三、完整代码

import java.util.*;


public class Solution {

    int[] dx = {0, 0, 1, -1};
    int[] dy = {1, -1, 0, 0};
    boolean[][] vis;
    int m, n;

    public int solve (char[][] grid) {
        // write code here
        m = grid.length;
        n = grid[0].length;
        vis = new boolean[m + 1][n + 1];
        int ret = 0;

        for(int i = 0; i <m; i++){
            for(int j = 0; j < n; j++){
                if(!vis[i][j] && grid[i][j] == '1'){
                    ret++;
                    dfs(grid, i, j);
                }
            }
        }

        return ret;
    }

    public void dfs(char[][] grid, int i, int j){
        vis[i][j] = true;
        for(int k = 0; k < 4; k++){
            int x = i + dx[k];
            int y = j + dy[k];
            if(x >= 0 && x < m && y >= 0 && y < n && !vis[x][y] && grid[x][y] == '1'){
                dfs(grid, x, y);
            }
        }
    }
}

以上就是本篇博客的全部内容啦,如有不足之处,还请各位指出,期待能和各位一起进步!