题目:
给定一个由 '1'
(陆地)和 '0'
(水)组成的的二维网格,计算岛屿的数量。一个岛被水包围,并且它是通过水平方向或垂直方向上相邻的陆地连接而成的。你可以假设网格的四个边均被水包围。
示例 1:
输入: 11110 11010 11000 00000 输出: 1
示例 2:
输入: 11000 11000 00100 00011 输出: 3
思路:
dfs,对每一个点进行dfs。
注意,测试数据有空数组的情况。
代码如下:
class Solution {
public:
int vis[1005][1005];
int loc[4][2]={{0,1},{0,-1},{1,0},{-1,0}};
int xSize,ySize;
bool Judge(int x,int y)
{
return x>=0&&x<xSize&&y>=0&&y<ySize;
}
void dfs (int x,int y,vector<vector<char>>& v)
{
vis[x][y]=1;
for (int i=0;i<4;i++)
{
int tx=x+loc[i][0];
int ty=y+loc[i][1];
if(Judge(tx,ty)&&vis[tx][ty]==0&&v[tx][ty]=='1')
{
dfs (tx,ty,v);
}
}
}
int numIslands(vector<vector<char>>& grid) {
int ans=0;
memset (vis,0,sizeof(vis));
xSize=grid.size();
if(xSize==0)
return 0;
ySize=grid[0].size();
if(ySize==0)
return 0;
for (int i=0;i<xSize;i++)
{
for (int j=0;j<ySize;j++)
{
if(!vis[i][j]&&grid[i][j]=='1')
{
dfs(i,j,grid);
ans++;
}
}
}
return ans;
}
};