200.岛屿数量

题目描述:

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

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

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

示例 1:

输入:
11110
11010
11000
00000
输出: 1
示例 2:

输入:
11000
11000
00100
00011
输出: 3
解释: 每座岛屿只能由水平和/或竖直方向上相邻的陆地连接而成。

思想:

深度优先搜索:当[i,j]处为'1'时,增加岛屿数,然后使用深度优先搜索将[i,j]附近为1的grid全部置为'0'

代码:

class Solution {
private:
    void dfs(vector<vector<char>>& grid,int i,int j){   //注意没有返回值用void,并且grid要引用
        int cols = grid[0].size();
        int rows = grid.size();
        grid[i][j] = '0';
        if(i-1>=0&&grid[i-1][j]=='1') dfs(grid,i-1,j);
        if(i+1<rows&&grid[i+1][j]=='1') dfs(grid,i+1,j);
        if(j-1>=0&&grid[i][j-1]=='1') dfs(grid,i,j-1);
        if(j+1<cols&&grid[i][j+1]=='1') dfs(grid,i,j+1);
    }
public:
    int numIslands(vector<vector<char>>& grid) {
        int count = 0;
        int rows = grid.size();
        if(rows == 0)   //注意判断输入为空的情况
            return count;
        int cols = grid[0].size();
        for(int i = 0;i < rows; i++)
            for(int j = 0;j < cols;j++){
                if(grid[i][j] == '1'){
                    count++;
                    dfs(grid,i,j);
                }
            }
        return count;
    }
};

猜你喜欢

转载自www.cnblogs.com/thefatcat/p/12743723.html