岛屿数量
岛屿数量
http://www.nowcoder.com/questionTerminal/0c9664d1554e466aa107d899418e814e
题目描述
给一个01矩阵,1代表是陆地,0代表海洋, 如果两个1相邻,那么这两个1属于同一个岛。我们只考虑上下左右为相邻。
岛屿: 相邻陆地可以组成一个岛屿(相邻:上下左右) 判断岛屿个数。
解法
//解法:深度优先搜索
// 逐点进行DFS. 一次DFS即搜索所有连通的1,并把1翻转为0,防止下次重复搜索。
// 时间O(N*M) 空间(N*M)
int solve(vector<vector<char> >& grid) {
int nr = grid.size();
if (nr == 0) 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;
}
void dfs(vector<vector<char> >& grid, int r, int c) {
int nr = grid.size();
int nc = grid[0].size();
if (r >= nr || c >= nc || grid[r][c] == '0' ) return ;
//搜索过的1,翻转为0, 防止重复计算
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);
return ;
}
OPPO成长空间 955人发布