岛屿数量
岛屿数量
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 ; }