[DFS BFS] 200. 岛屿数量(DFS + used数组 + 空间优化)

200. 岛屿数量

题目链接:https://leetcode-cn.com/problems/number-of-islands/

相似题目:130. 被围绕的区域

分类:深度优先遍历、广度优先遍历、图、并查集(待补充)

在这里插入图片描述

思路1:DFS + 辅助数组 -> 空间优化

设置一个辅助二维数组记录每个元素是否被访问过。
算法运行的每一轮是在寻找一个岛屿,如果在寻找过程中发现某一元素被访问过,说明这个元素已经被记录为其他岛屿了,不能作为当前岛屿的节点。

关键问题:如何寻找当前岛屿的所有节点?

可以用递归回溯来实现。
每个节点都枚举自己上下左右相邻的节点,如果节点值为1且未被访问过,则置为访问过,然后以该节点为起点,重新判断它的上下左右。

如果上下左右全为0或已访问过的1,则直接return.

算法流程
  1. 设置一个辅助二维数组来记录节点是否被访问过,count计数器用来记录岛屿数量。
  2. 遍历grid数组,寻找每个岛屿的起点,如果某个节点=='1’且未被访问过,就可以作为一个新岛屿的起点,count+1,然后以该点为起点调用递归函数findIsland将同在一个岛屿上的其他节点全部置为“访问过”。
  3. 最终count存放的就是岛屿数量。
实现代码
class Solution {
    public int numIslands(char[][] grid) {
        //特殊用例:null , [], [[]]
        if(grid == null || grid.length == 0 || grid[0].length == 0) return 0;
        int rows = grid.length, cols = grid[0].length;
        boolean[][] used = new boolean[rows][cols];
        int count = 0;//记录岛屿的数量
        //遍历整个grid数组,寻找岛屿的起点
        for(int i = 0; i < rows; i++){
            for(int j = 0; j < cols; j++){
                //找到一个起点
                if(grid[i][j] == '1' && !used[i][j]){
                    count++;//岛屿数量加1
                    findIsland(i, j, grid, used);//访问所有同一岛屿上的节点
                }
            }
        }
        return count;
    }
    //以一个节点为起点,寻找所在岛屿的所有节点
    public void findIsland(int x, int y, char[][] grid, boolean[][] used){
        if(grid[x][y] == '1' && !used[x][y]) 
            used[x][y] = true;
        else return;
        //上
        if(y - 1 >= 0) findIsland(x, y - 1, grid, used);
        //下
        if(y + 1 < grid[0].length) findIsland(x, y + 1, grid, used);
        //左
        if(x - 1 >= 0) findIsland(x - 1, y, grid, used);
        //右
        if(x + 1 < grid.length) findIsland(x + 1, y, grid, used);
    } 
}
  • 时间复杂度:O((MN)^2),
  • 空间复杂度:O(MN)

空间优化:可以借用grid本身作为used记录节点是否被访问过,因为grid的元素只有两种取值’0’或’1’,所以可以设置’2’表示该节点被访问过,即可不使用used辅助数组,将空间复杂度降为O(1).

  • 当然能做这个优化的前提是允许修改输入数据。

实现代码

class Solution {
    public int numIslands(char[][] grid) {
        //特殊用例:null , [], [[]]
        if(grid == null || grid.length == 0 || grid[0].length == 0) return 0;
        int rows = grid.length, cols = grid[0].length;
        int count = 0;//记录岛屿的数量
        //遍历整个grid数组,寻找岛屿的起点
        for(int i = 0; i < rows; i++){
            for(int j = 0; j < cols; j++){
                //找到一个起点
                if(grid[i][j] == '1'){
                    count++;//岛屿数量加1
                    findIsland(i, j, grid);//访问所有同一岛屿上的节点
                }
            }
        }
        return count;
    }
    //以一个节点为起点,寻找所在岛屿的所有节点
    public void findIsland(int x, int y, char[][] grid){
        if(grid[x][y] == '1') 
            grid[x][y] = '2';
        else return;
        //上
        if(y - 1 >= 0) findIsland(x, y - 1, grid);
        //下
        if(y + 1 < grid[0].length) findIsland(x, y + 1, grid);
        //左
        if(x - 1 >= 0) findIsland(x - 1, y, grid);
        //右
        if(x + 1 < grid.length) findIsland(x + 1, y, grid);
    } 
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值