题目描述:
给定一个由 '1'(陆地)和 '0'(水)组成的的二维网格,计算岛屿的数量。一个岛被水包围,并且它是通过水平方向或垂直方向上相邻的陆地连接而成的。你可以假设网格的四个边均被水包围。
示例 1:
输入:
11110
11010
11000
00000
输出: 1
示例 2:
输入:
11000
11000
00100
00011
输出: 3
思路:求图中连通分量的个数,按照深度优先遍历或者广度优先遍历,遍历整个图,找到有几个连通分量即可
//代码贴的leetcode的官方思路
class Solution {
private:
void dfs(vector<vector<char>>& grid, int r, int c) {
int nr = grid.size();
int nc = grid[0].size();
grid[r][c] = '0';
if (r - 1 >= 0 && grid[r-1][c] == '1') dfs(grid, r - 1, c);//左边一个点
if (r + 1 < nr && grid[r+1][c] == '1') dfs(grid, r + 1, c);//右边一个点
if (c - 1 >= 0 && grid[r][c-1] == '1') dfs(grid, r, c - 1);//上面一个点
if (c + 1 < nc && grid[r][c+1] == '1') dfs(grid, r, c + 1);//下面一个点
}
public:
int numIslands(vector<vector<char>>& grid) {
int nr = grid.size();
if (!nr) return 0;
int nc = grid[0].size();
int num_islands = 0;
for (int r = 0; r < nr; ++r) {
for (int c = 0; c < nc; ++c) {
if (grid[r][c] == '1') {
++num_islands;
dfs(grid, r, c);
}
}
}
return num_islands;
}
};
作者自己的代码
class Solution {
public:
int count=0;
int X[4]={0,0,1,-1};
int Y[4]={1,-1,0,0};
void dfs(vector<vector<char>>& grid,int p,int q,int row,int col){
grid[p][q]='0';
for(int k=0;k<4;k++){
int newx=p+X[k];
int newy=q+Y[k];
if(newx>=0&&newx<row&&newy>=0&&newy<col&&grid[newx][newy]=='1')
dfs(grid,newx,newy,row,col);
}
}
int numIslands(vector<vector<char>>& grid) {
int row=grid.size();
int col=grid[0].size();
for(int i=0;i<row;i++){
for(int j=0;j<col;j++){
if(grid[i][j]=='1'){
count++;
dfs(grid,i,j,row,col);
}
}
}
return count;
}
};