给定一个由 ‘1’(陆地)和 ‘0’(水)组成的的二维网格,计算岛屿的数量。一个岛被水包围,并且它是通过水平方向或垂直方向上相邻的陆地连接而成的。你可以假设网格的四个边均被水包围。
示例 1:
输入:
11110
11010
11000
00000
输出: 1
示例 2:
输入:
11000
11000
00100
00011
输出: 3
class Solution {
private:
void dfs(vector<vector<char>>& grid, int x, int y, vector<int> &dx, vector<int> &dy)
{
grid[x][y] = '0';
int rowSize = grid.size();
int colSize = grid[0].size();
for(int i = 0; i < 4; i++)
{
int tx = x + dx[i];
int ty = y + dy[i];
if(tx < 0 || tx >= rowSize || ty < 0 || ty >= colSize)
continue;
if(grid[tx][ty] == '1')
{
dfs(grid, tx, ty, dx, dy);
}
}
}
public:
int numIslands(vector<vector<char>>& grid) {
if(grid.size() == 0 || grid[0].size() == 0) return 0;
int row = grid.size();
int col = grid[0].size();
int numRes = 0;
vector<int> dx = {-1, 1, 0, 0};
vector<int> dy = {0, 0, -1, 1};
for(int i = 0; i < row; i++)
{
for(int j = 0; j < col; j++)
{
if(grid[i][j] == '1')
{
numRes++;
dfs(grid, i, j, dx, dy);
}
}
}
return numRes;
}
};
首先,开始先遍历地图,直到遇到第一个‘1’(岛屿),以它为root开始通过DFS搜索这座岛屿的其它部分,并把他们全部转化为‘0’,同时把岛屿数量+1。之后接着遍历地图,由于我们将搜索过的岛屿转化为‘0’,因此不会重复搜索。