【leetcode】200 岛屿数量（DFS）

最新推荐文章于 2022-11-29 20:24:36 发布

zjwreal

最新推荐文章于 2022-11-29 20:24:36 发布

阅读量372

点赞数

分类专栏： LeetCode 文章标签： DFS

本文链接：https://blog.csdn.net/zjwreal/article/details/94174092

版权

LeetCode 专栏收录该内容

297 篇文章 3 订阅

订阅专栏

题目链接：

题目描述

思路

用一个二维数字visited标记节点是否被访问过；
DFS遍历相连通的’1’节点进行标记

复杂度分析
时间复杂度：O（n^2）
空间复杂度：O（n^2）

class Solution {
public:
    int numIslands(vector<vector<char>>& grid) {
        if(grid.empty() || grid[0].empty()) return 0;
        int rows = grid.size(), cols = grid[0].size();
        vector<vector<bool>> visited(rows,vector<bool>(cols, false));
        int ret = 0;
        for (int i = 0; i < rows; ++i) {
            for (int j = 0; j < cols; ++j) {
                if(!visited[i][j] && grid[i][j] == '1'){
                    ret ++;
                    dfs(grid,visited,i,j);
                }
            }
        }
        return ret;
    }

private:
    // 深度优先搜索将与该位置相连的‘1’节点全部标记为访问过
    void dfs(vector<vector<char>>& grid, vector<vector<bool>>& visited, int row, int col){
        if(row < 0 || col < 0 || row >= grid.size() || col>=grid[0].size() || visited[row][col]) return;
        if(grid[row][col] == '1') {
            visited[row][col] = true;
            dfs(grid,visited,row-1,col);
            dfs(grid,visited,row+1,col);
            dfs(grid,visited,row,col-1);
            dfs(grid,visited,row,col+1);
        }
    }
};

在这里插入图片描述

进一步优化，直接在grid矩阵上标记，空间复杂度降为O（1）

class Solution {
public:
    int numIslands(vector<vector<char>>& grid) {
        if(grid.empty() || grid[0].empty()) return 0;
        int rows = grid.size(), cols = grid[0].size();
        vector<vector<bool>> visited(rows,vector<bool>(cols, false));
        int ret = 0;
        for (int i = 0; i < rows; ++i) {
            for (int j = 0; j < cols; ++j) {
                if(grid[i][j] == '1'){
                    ret ++;
                    dfs(grid,i,j);
                }
            }
        }
        return ret;
    }

private:
    // 深度优先搜索将与该位置相连的‘1’节点全部标记为访问过
    void dfs(vector<vector<char>>& grid,  int row, int col){
        if(row < 0 || col < 0 || row >= grid.size() || col>=grid[0].size() || grid[row][col] == '2') return;
        if(grid[row][col] == '1') {
            grid[row][col] = '2';
            dfs(grid,row-1,col);
            dfs(grid,row+1,col);
            dfs(grid,row,col-1);
            dfs(grid,row,col+1);
        }
    }
};