岛屿问题DFS(深度优先遍历)
给你一个由 '1'(陆地)和 '0'(水)组成的的二维网格,请你计算网格中岛屿的数量。
岛屿总是被水包围,并且每座岛屿只能由水平方向和/或竖直方向上相邻的陆地连接形成。
此外,你可以假设该网格的四条边均被水包围。
输入
输入:grid = [
["1","1","1","1","0"],
["1","1","0","1","0"],
["1","1","0","0","0"],
["0","0","0","0","0"]
]
输出:1
对于这类题目利用DFS还是比较好理解,实际上就是去找为’1’的陆地,先遍历找到第一个陆地之后,然后从第一个向上下左右继续遍历,直到为四周为0停止,这样就是一块陆地
如下图。
void dfs(vector<vector<char>>&grid,int l,int r)
{
int ls=grid.size();
int rs=grid[0].size();
grid[l][r]='0';//先标记把他化成海洋
if(l-1>=0&&grid[l-1][r]=='1') dfs(grid,l-1,r);
if(l+1<ls&&grid[l+1][r]=='1') dfs(grid,l+1,r);
if(r-1>=0&&grid[l][r-1]=='1') dfs(grid,l,r-1);
if(r+1<rs&&grid[l][r+1]=='1') dfs(grid,l,r+1);
}
一般DFS的函数就是这样,也就是上图中第一个数开始遍历直到变为周围全是0,停止。
防止重复遍历所以遍历过的1就给他为0,这样到最后一个数周围全是0
grid[l][r]='0';//先标记把他化成海洋
完整代码
int numIslands(vector<vector<char>>& grid) {
int ls=grid.size();
int rs=grid[0].size();
int sum=0;
for(int i=0;i<ls;i++)
for(int j=0;j<rs;j++)
{
if(grid[i][j]=='1')
{sum++;
dfs(grid,i,j);
}
}
return sum;
}
//DFS
void dfs(vector<vector<char>>&grid,int l,int r)
{
int ls=grid.size();
int rs=grid[0].size();
grid[l][r]='0';//先标记把他化成海洋
if(l-1>=0&&grid[l-1][r]=='1') dfs(grid,l-1,r);
if(l+1<ls&&grid[l+1][r]=='1') dfs(grid,l+1,r);
if(r-1>=0&&grid[l][r-1]=='1') dfs(grid,l,r-1);
if(r+1<rs&&grid[l][r+1]=='1') dfs(grid,l,r+1);
}
};