Given a 2d grid map of '1'
s (land) and '0'
s (water), count the number of islands. An island is surrounded by water and is formed by connecting adjacent lands horizontally or vertically. You may assume all four edges of the grid are all surrounded by water.
Example 1:
Input: 11110 11010 11000 00000 Output: 1
class Solution {
public:
int numIslands(vector<vector<char>>& grid) {
if(grid.empty())
return 0;
int m=grid.size();
int n=grid[0].size();
int len=0;
for(int i=0;i<m;i++)
for(int j=0;j<n;j++)
{
/*if(grid[i][j]=='1')
{
grid[i][j]='0';
search(grid,i,j);
len++;
}*/
len+=grid[i][j]-'0';
dfs(grid,i,j,m,n);
}
return len;
}
void dfs(vector<vector<char>>& grid,int i,int j,int m,int n){
if(i>=m||j>=n||i<0||j<0||grid[i][j]=='0')
return;
grid[i][j]='0';
dfs(grid,i+1,j,m,n);
dfs(grid,i-1,j,m,n);
dfs(grid,i,j+1,m,n);
dfs(grid,i,j-1,m,n);
}
本题最重要的一点是,找到一个‘1’之后,去找和它相连的‘1’,并把找到得这些‘1’全部置零。这是因为一个DFS结束,就会去找下一个岛(等于‘1’的点),而把这些‘1’全部置零会避免下次找到的‘1’是在已经发现的小岛上。
深度遍历DFS结束的条件是,数组越界,或者grid[i][j]=='0'
一定要注意:在if(i>=m||j>=n||i<0||j<0||grid[i][j]=='0'),grid[i][j]=='0'这个条件一定要放在最后,我一开始放在最前面,leetcode会报错:AddressSanitizer: heap-buffer-overflow...这个错误是说你访问数组时越界了。所以要先判断i,j索引是否越界,在不越界的情况下再去访问。
还有另外一个问题:一开始我认为每个‘1’只需要去判断和它右相邻和下相邻的点即可,左边和上面已经判断过了。但真实情况不是这样。因为只是在最外层循环找新岛的时候,是按照向下和向右的,在内层循环去扩张岛的时候,某个点(i,j)的左邻点(i-1,j)和上邻点(i,j-1)不一定已经被判断过了(可能这两个点的左邻点(i-2,j),(j-1,j-1)或者上邻点(i-1,j-1),(j,j-2)是0,导致后续相邻点没有去做判断)因此,还是要从上下左右四个方向去做深度遍历的。
例子:
1 1 1
0 1 0
1 1 1
如果按我那种方法,左下角的1会被判断成另外一个岛。
一开始我写的深度搜索函数是这样的:找到一个为1 的点以后(这是前提),判断(i,j)这个点的相邻点是否为1,然后再递归,但这样很麻烦,也没处理好i+1,j+1的越界问题。而改进之后的函数dfs,没有了前提,会先判断该点是否为1,为0就直接返回。就这点差别也使得代码简洁了许多。
void search(vector<vector<char>>& grid,int i,int j){
if(grid[i+1][j]=='0'&&grid[i][j+1]=='0')
return;
if(grid[i+1][j]=='1')
{
grid[i+1][j]='0';
if(i+1==grid.size()-1)
return;
search(grid,i+1,j);
}
if(grid[i][j+1]=='1')
{
grid[i][j+1]='0';
if(j+1==grid[0].size()-1)
return;
search(grid,i,j+1);
}
}