计算岛屿的数量

最新推荐文章于 2024-03-29 11:52:53 发布

原创最新推荐文章于 2024-03-29 11:52:53 发布 · 662 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#数据结构

program 专栏收录该内容

15 篇文章

订阅专栏

该博客讨论了一种计算二维数组中岛屿数量的方法。数组中的1代表岛屿，0代表海水，相邻的1视为同一岛屿。通过深度优先搜索（DFS）遍历数组，当遇到1时增加岛屿计数，并将已访问的1置为0。提供的示例代码展示了如何实现这个算法，最终返回岛屿的数量。

给一个二维数组,元素只有0和1,1代表岛屿,0代表海水,如果与1相邻的位置还是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三个
*/

上述例子中就只包含了三个岛屿.

解题思路:如果遍历到当前元素是1,岛屿数量+1,然后在深度优先遍历它上下左右四个位置的元素,把与它相邻且为1的元素都置为0

int solve(std::vector<std::vector<char> >& grid) {
    // write code here
    int cnt = 0;
    for(int i = 0; i < grid.size(); i++){
        for(int j = 0; j < grid[0].size(); j++){
            if(grid[i][j] == '1'){
                cnt++;
                dfs(grid, i, j);
            }
        }
    }
    return cnt;
}

void dfs(std::vector<std::vector<char> >& grid, int x, int y) {
    //上下右四个方向
    int dx[] = {0, 0, -1, 1};
    int dy[] = {-1, 1, 0, 0};
    grid[x][y] = '0'; //走过的路记0
    for(int i = 0; i < 4; i++){
        int tx = x + dx[i];
        int ty = y + dy[i];
        //超出边界
        if(tx < 0 || ty < 0 || tx >= grid.size() || ty >= grid[0].size()){
            continue;
        }
        if(grid[tx][ty] == '1'){
            dfs(grid, tx, ty);
        }
    }
}