题目描述
给定一个由 '1'
(陆地)和 '0'
(水)组成的的二维网格,计算岛屿的数量。一个岛被水包围,并且它是通过水平方向或垂直方向上相邻的陆地连接而成的。你可以假设网格的四个边均被水包围。
示例 1:
输入:
11110
11010
11000
00000
输出: 1
示例 2:
输入:
11000
11000
00100
00011
输出: 3
题目难度:★★★★
题目地址
https://leetcode-cn.com/problems/number-of-islands
解决思路
采用DFS深度优先搜索,创建visited保存比那里过的位置。采用递归遍历数组中与1关联的位置。
解决方案
class Solution {
public:
int numIslands(vector<vector<char>>& grid) {
if (grid.empty() || grid[0].empty()) return 0;
int m = grid.size(), n = grid[0].size(), res = 0;
vector<vector<bool>> visited(m, vector<bool>(n,false));
for (int i = 0; i < m; ++i) {
for (int j = 0; j < n; ++j) {
if (grid[i][j] == '0' || visited[i][j]) continue;
numIslandsDFS(grid, visited, i, j);
++res;
}
}
return res;
}
void numIslandsDFS(vector<vector<char>> & grid, vector<vector<bool>> & visited, int x, int y) {
if (x < 0 || x >= grid.size() || y < 0 || y >= grid[0].size() || grid[x][y] == '0' || visited[x][y]) return;
visited[x][y] = true;
numIslandsDFS(grid, visited, x - 1, y);
numIslandsDFS(grid, visited, x + 1, y);
numIslandsDFS(grid, visited, x, y - 1);
numIslandsDFS(grid, visited, x, y + 1);
}
};
采用DFS,对二维数组进行一次遍历,遇到1则找到一个岛屿,然后把与之关联的位置置2即可。
class Solution {
public:
void numIslandsDFS(vector<vector<char>> &grid,int row,int col){
int m=grid.size();
int n=grid[0].size();
if((row<0)||(row>=m)||(col<0)||(col>=n)||(grid[row][col]!='1')){
return;
}
grid[row][col]=2;
numIslandsDFS(grid,row+1,col);
numIslandsDFS(grid,row-1,col);
numIslandsDFS(grid,row,col+1);
numIslandsDFS(grid,row,col-1);
}
int numIslands(vector<vector<char>>& grid) {
int m=grid.size();
if(m==0) return 0;
int n=grid[0].size();
if(n==0) return 0;
int sum=0;
for(int i=0;i<m;i++){
for(int j=0;j<n;j++){
if(grid[i][j]=='1'){
sum++;
setOne(grid,i,j);
}
}
}
return sum;
}
};