C++ 计算岛屿数量

/*
给一个01矩阵,1代表是陆地,0代表海洋,如果两个1相邻,那么这两个1属于同一个岛。我们只考虑上下左右为相邻。
岛屿:相邻陆地可以组成一个岛屿(相邻:上下左右)判断岛屿个数。
输入描述:01矩阵
输出描述:岛屿个数
样例输入:[[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
*/

#include <iostream>
#include <vector>

using namespace std;

// 定义上下左右四个方向的偏移量
int dx[] = { -1, 1, 0, 0 };
int dy[] = { 0, 0, -1, 1 };

// 检查坐标是否在矩阵范围内
bool isValid(int x, int y, int numRows, int numCols) {
    return x >= 0 && x < numRows && y >= 0 && y < numCols;
}

// 深度优先搜索,将属于同一个岛的陆地标记为已访问
void dfs(vector<vector<int>>& grid, int x, int y) {
    int numRows = grid.size();
    int numCols = grid[0].size();

    // 标记当前陆地为已访问
    grid[x][y] = 0;

    // 检查上下左右四个方向的相邻陆地
    for (int i = 0; i < 4; i++) {
        int newX = x + dx[i];
        int newY = y + dy[i];

        // 如果相邻坐标合法且是陆地,则继续深度优先搜索
        if (isValid(newX, newY, numRows, numCols) && grid[newX][newY] == 1) {
            dfs(grid, newX, newY);
        }
    }
}

int numIslands(vector<vector<int>>& grid) {
    int numRows = grid.size();
    int numCols = grid[0].size();
    int numIslands = 0;

    for (int i = 0; i < numRows; i++) {
        for (int j = 0; j < numCols; j++) {
            if (grid[i][j] == 1) {
                // 发现新的陆地,开始深度优先搜索并增加岛屿计数
                numIslands++;
                dfs(grid, i, j);
            }
        }
    }

    return numIslands;
}

int main() {
    vector<vector<int>> grid = { {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} };

    int result = numIslands(grid);
    cout << "岛屿个数为: " << result << endl;

    return 0;
}

全部评论

相关推荐

头像
11-21 11:39
四川大学 Java
是红鸢啊:忘了还没结束,还有字节的5k 违约金
点赞 评论 收藏
分享
评论
点赞
1
分享
牛客网
牛客企业服务