描述
给一个01矩阵,求不同的岛屿的个数。
0代表海,1代表岛,如果两个1相邻,那么这两个1属于同一个岛。我们只考虑上下左右为相邻。
您在真实的面试中是否遇到过这个题?
是
样例
在矩阵:
[
[1, 1, 0, 0, 0],
[0, 1, 0, 0, 1],
[0, 0, 0, 1, 1],
[0, 0, 0, 0, 0],
[0, 0, 0, 0, 1]
]
中有 3
个岛.
深度优先搜索的例题,循环遍历,找到岛屿后,进行深度优先搜索,将其置零。
class Solution {
public:
/**
* @param grid a boolean 2D matrix
* @return an integer
*/
int numIslands(vector<vector<bool>>& grid) {
// Write your code here
if(grid.empty() || grid[0].empty())
return 0;
int m = grid.size(), n = grid[0].size();
int count = 0;
for(int i = 0; i < m; ++i){
for(int j = 0; j < n; ++j){
if(grid[i][j]){
++count;
dfs(grid, i, j);
}
}// for
}// for
return count;
}// numIslands
void dfs(vector<vector<bool> > &grid, int x, int y){
if((x < 0) || (y < 0) || (x >= grid.size()) || (y >= grid[0].size()) || !grid[x][y])
return;
grid[x][y] = false;
dfs(grid, x - 1, y);
dfs(grid, x, y - 1);
dfs(grid, x + 1, y);
dfs(grid, x, y + 1);
}
};